|
| 1 | +<!--?title Finding all sub-palindromes in O(N)--> |
| 2 | + |
| 3 | +# Finding all sub-palindromes in $O(N)$ |
| 4 | + |
| 5 | +## Statement |
| 6 | + |
| 7 | +Given string $s$ with length $n$. Find all the pairs $(i, j)$ such that substring $s[i\dots j]$ is a palindrome. String $t$ is a palindrome when $t = t_{rev}$ ($t_{rev}$ is a reversed string for $t$). |
| 8 | + |
| 9 | +## More precise statement |
| 10 | + |
| 11 | +It's clear that in the worst case we can have $O(n^2)$ palindrome strings, and at the first glance it seems that there is no linear algorithm for this problem. |
| 12 | + |
| 13 | +But the information about the palindromes can be kept **in a more compact way**: for each position $i = 0\dots n-1$ we'll find the values $d_1[i]$ and $d_2[i]$, denoting the number of palindromes accordingly with odd and even lengths with centers in the position $i$. |
| 14 | + |
| 15 | +For instance, string $s = abababc$ has three palindromes with odd length with centers in the position $s[3] = b$, i. e. $d_1[3] = 3$: |
| 16 | + |
| 17 | +$a\ \overbrace{b\ a\ \underbrace{b}_{s_3}\ a\ b}^{d_1[3]=3} c$ |
| 18 | + |
| 19 | +And string $s = cbaabd$ has two palindromes with even length with centers in the position $s[3] = a$, i. e. $d_2[3] = 2$: |
| 20 | + |
| 21 | +$c\ \overbrace{b\ a\ \underbrace{a}_{s_3}\ b}^{d_2[3]=2} d$ |
| 22 | + |
| 23 | +So the idea is that if we have a sub-palindrome with length $l$ with center in some position $i$, we also have sub-palindromes with lengths $l-2$, $l-4$ etc. with centers in $i$. So these two arrays $d_1[i]$ and $d_2[i]$ are enough to keep the information about all the sub-palindromes in the string. |
| 24 | + |
| 25 | +It's a surprising fact that there is an algorithm, which is simple enough, that calculates these "palindromity arrays" $d_1[]$ and $d_2[]$ in linear time. The algorithm is described in this article. |
| 26 | + |
| 27 | +## Solution |
| 28 | + |
| 29 | +In general, this problem has many solutions: with [String Hashing](/string/string-hashing.html) it can be solved in $O(n\cdot \log n)$, and with [Suffix Trees](/string/suffix-tree-ukkonen.html) and fast LCA this problem can be solved in $O(n)$. |
| 30 | + |
| 31 | +But the method described here is **sufficiently** simpler and has less hidden constant in time and memory complexity. This algorithm was discovered by **Glenn K. Manacher** in 1975. |
| 32 | + |
| 33 | +## Trivial algorithm |
| 34 | + |
| 35 | +To avoid ambiguities in the further description we denote what "trivial algorithm" is. |
| 36 | + |
| 37 | +It's the algorithm that does the following. For each center position $i$ it tries to increase the answer by one until it's possible, comparing a pair of corresponding characters each time. |
| 38 | + |
| 39 | +Such algorithm is slow, it can calculate the answer only in $O(n^2)$. |
| 40 | + |
| 41 | +The implementation of the trivial algorithm is: |
| 42 | + |
| 43 | +```cpp |
| 44 | +vector<int> d1(n), d2(n); |
| 45 | +for (int i = 0; i < n; i++) { |
| 46 | + d1[i] = 1; |
| 47 | + while (0 <= i - d1[i] && i + d1[i] < n && s[i - d1[i]] == s[i + d1[i]]) { |
| 48 | + d1[i]++; |
| 49 | + } |
| 50 | + |
| 51 | + d2[i] = 0; |
| 52 | + while (0 <= i - d2[i] - 1 && i + d2[i] < n && s[i - d2[i] - 1] == s[i + d2[i]]) { |
| 53 | + d2[i]++; |
| 54 | + } |
| 55 | +} |
| 56 | +``` |
| 57 | +
|
| 58 | +## Manacher's algorithm |
| 59 | +
|
| 60 | +We describe the algorithm to find all the sub-palindromes with odd length, i. e. to calculate $d_1[]$; the solution for all the sub-palindromes with even length (i. e. calculating the array $d_2[]$) will be a minor modification for this one. |
| 61 | +
|
| 62 | +For fast calculation we'll keep the **borders $(l, r)$** of the rightmost found sub-palindrome (i. e. the palindrome with maximal $r$). Initially we assume $l = 0, r = -1$. |
| 63 | +
|
| 64 | +So, we want to calculate $d_1[i]$ for the next $i$, and all the previous values in $d_1[]$ have been already calculated. We do the following: |
| 65 | +
|
| 66 | +* If $i$ is outside the current sub-palindrome, i. e. $i > r$, we'll just launch the trivial algorithm. |
| 67 | + |
| 68 | + So we'll increase $d_1[i]$ consecutively and check each time if the current substring $[i - d_1[i]\dots i + d_1[i]]$ is a palindrome. When we find first divergence or meet the boundaries of $s$, we'll stop. In this case we've finally calculated $d_1[i]$. After this, we must not forget to update $(l, r)$. |
| 69 | +
|
| 70 | +* Now consider the case when $i \le r$. We'll try to extract some information from the already calculated values in $d_1[]$. So, let's flip the position $i$ inside the sub-palindrome $(l, r)$, i. e. we'll get the position $j = l + (r - i)$, and we'll look on the value $d_1[j]$. Because $j$ is the position symmetrical to $i$, we'll **almost always** can assign $d_1[i] = d_1[j]$. Illustration of this (palindrome around $j$ is actually "copied" into the palindrome around $i$): |
| 71 | + |
| 72 | + $$ |
| 73 | + \ldots\ |
| 74 | + \overbrace{ |
| 75 | + s\_l\ \ldots\ |
| 76 | + \underbrace{ |
| 77 | + s\_{j-d_1[j]+1}\ \ldots\ s_j\ \ldots\ s\_{j+d_1[j]-1}\ |
| 78 | + }\_\text{palindrome}\ |
| 79 | + \ldots\ |
| 80 | + \underbrace{ |
| 81 | + s\_{i-d_1[j]+1}\ \ldots\ s_i\ \ldots\ s\_{i+d_1[j]-1}\ |
| 82 | + }\_\text{palindrome}\ |
| 83 | + \ldots\ s_r\ |
| 84 | + }^\text{palindrome}\ |
| 85 | + \ldots |
| 86 | + $$ |
| 87 | + |
| 88 | + But there is a **tricky case** to be handled correctly: when the "inner" palindrome reaches the borders of the "outer" one, i. e. $j - d_1[j] + 1 \le l$ (or, which is the same, $i + d_1[j] - 1 \ge r$). Because the symmetry outside the "outer" palindrome is not guaranteed, just assigning $d_1[i] = d_1[j]$ will be incorrect: we have not enough data to state that the palindrome in the position $i$ has the same length. |
| 89 | + |
| 90 | + Actually, we should "cut" the length of our palindrome, i. e. assign $d_1[i] = r - i$, to handle such situations correctly. After this we'll run the trivial algorithm which will try to increase $d_1[i]$ while it's possible. |
| 91 | + |
| 92 | + Illustration of this case (the palindrome with center $j$ is already "cut" to fit the "outer" palindrome): |
| 93 | + |
| 94 | + $$ |
| 95 | + \ldots\ |
| 96 | + \overbrace{ |
| 97 | + \underbrace{ |
| 98 | + s_l\ \ldots\ s_j\ \ldots\ s_{j+(j-l)}\ |
| 99 | + }\_\text{palindrome}\ |
| 100 | + \ldots\ |
| 101 | + \underbrace{ |
| 102 | + s_{i-(r-i)}\ \ldots\ s_i\ \ldots\ s_r |
| 103 | + }\_\text{palindrome}\ |
| 104 | + }^\text{palindrome}\ |
| 105 | + \underbrace{ |
| 106 | + \ldots \ldots \ldots \ldots \ldots |
| 107 | + }\_\text{try moving here} |
| 108 | + $$ |
| 109 | + |
| 110 | + It is shown on the illustration that, though the palindrome with center $j$ could be larger and go outside the "outer" palindrome, in the position $i$ we can use only the part that entirely fits into the "outer" palindrome. But the answer for the position $i$ can be much longer that this part, so next we'll run our trivial algorithm that will try to grow it outside our "outer" palindrome, i. e. to the region "try moving here". |
| 111 | +
|
| 112 | +At the end, it's necessary to remind that we should not forget to update the values $(l, r)$ after calculating each $d_1[i]$. |
| 113 | +
|
| 114 | +Also we'll repeat that the algorithm was described to calculate the array for odd palindromes $d_1[]$, the algorithm is similar for the array of even palindromes $d_2[]$. |
| 115 | +
|
| 116 | +## Complexity of Manacher's algorithm |
| 117 | +
|
| 118 | +At the first glance it's not obvious that this algorithm has linear time complexity, because we often run the naive algorithm while searching the answer for a particular position. |
| 119 | +
|
| 120 | +But more careful analysis shows that the algorithm is linear however. We need to mention [Z-function building algorithm](/string/z-function.html) which looks similar to this algorithm and also works in linear time. |
| 121 | +
|
| 122 | +Actually, we can notice that every iteration of trivial algorithm makes $r$ increase by one. Also $r$ cannot be decreased during the algorithm. So, trivial algorithm will make $O(n)$ iterations in total. |
| 123 | +
|
| 124 | +Also, other parts of Manacher's algorithm work obviously in linear time, we get $O(n)$ time complexity. |
| 125 | +
|
| 126 | +## Implementation of Manacher's algorithm |
| 127 | +
|
| 128 | +For calculating $d_1[]$, we get the following code: |
| 129 | +
|
| 130 | +```cpp |
| 131 | +vector<int> d1(n); |
| 132 | +for (int i = 0, l = 0, r = -1; i < n; i++) { |
| 133 | + int k = (i > r) ? 1 : min(d1[l + r - i], r - i); |
| 134 | + while (0 <= i - k && i + k < n && s[i - k] == s[i + k]) { |
| 135 | + k++; |
| 136 | + } |
| 137 | + d1[i] = k--; |
| 138 | + if (i + k > r) { |
| 139 | + l = i - k; |
| 140 | + r = i + k; |
| 141 | + } |
| 142 | +} |
| 143 | +``` |
| 144 | + |
| 145 | +For calculating $d_2[]$, the code looks similar, but with minor changes in arithmetical expressions: |
| 146 | + |
| 147 | +```cpp |
| 148 | +vector<int> d2(n); |
| 149 | +for (int i = 0, l = 0, r = -1; i < n; i++) { |
| 150 | + int k = (i > r) ? 0 : min(d2[l + r - i + 1], r - i + 1); |
| 151 | + while (0 <= i - k - 1 && i + k < n && s[i - k - 1] == s[i + k]) { |
| 152 | + k++; |
| 153 | + } |
| 154 | + d2[i] = k--; |
| 155 | + if (i + k > r) { |
| 156 | + l = i - k - 1; |
| 157 | + r = i + k ; |
| 158 | + } |
| 159 | +} |
| 160 | +``` |
| 161 | +
|
| 162 | +# Problems |
| 163 | +
|
| 164 | +[UVA #11475 "Extend to Palindrome"](https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=2470) |
0 commit comments