午夜稻草人

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

回文串包括奇数长的和偶数长的,一般求的时候都要分情况讨论,这个算法做了个简单的处理把奇偶情况统一了。算法的基本思路是这样的,把原串每个字符中间用一个串中没出现过的字符分隔开来(统一奇偶),用一个数组p[ i ]记录以 str[ i ] 为中间字符的回文串向右能匹配的长度。先看个例子

原串:       w  a   a   b   w   s   w   f   d

新串(str):  #   w  #   a   #   a   #   b  #   w   #   s    #   w    #     f    #    d     #

               0   1   2   3   4    5   6   7   8   9  10  11 12  13  14   15  16   17   18

p数组:     1   2   1   2   3    2   1   2   1   2   1    4   1    2    1     2    1    2    1

由p数组的性质,新串中以str[i]为中间字符的回文串的长度为p[i]-1(可以对照p[11]这个位置,p[i]-1本身表示对称半径,但是实际上去掉#以后,p[i]-1就是回文串长度),以#为中间字符的就是长度为偶数的,以非#号为中间字符的就是长度为奇数的,那么怎么求p[ ]数组呢?

从左到右计算(0~str.length),也就是计算p[i]时,p[0.....i-1] 都已经计算出来了,并且用一个变量mx记录当前检测出的回文串的右侧最大位置 max{ k+p[ k ] } (k=0.....i-1),用id记录取最大值时的k。


上面的这个截图是很多人都用过的,需要注意的是, 两张图分别表示了当前点i<mx时的两种情况:

1) 当前点i关于id的对称点j, 以j为中心的回文串的左边界不小于id-p[id],根据回文串的对称性, 这就意味着i的回文串长度是跟j是一样的, 所以有p[i] = p[j] = p[2*id-i];

2) 如果以j为中心的回文串的左边界小于id-p[id],则只能确保p[i]>=mx-i, 至于p[i]的值具体为多少,还需要检测mx后面的位置才能确定出来。

所以就有了下面的这个关键代码,理解了这部分,整个算法就好理解了。

if( mx > i )

p[i] = MIN( p[2*id-i], mx-i );

 完整代码如下:

 

 1 #include<iostream>
 2 #include<string>
 3 #include<stdlib.h>
 4 using namespace std;
 5 
 6 char cArray[1000];
 7 int p[1000];
 8 
 9 int manacher(int length)
10 {
11     int mx = 0;
12     int id = 0;
13     int maxLength = 0;
14     
15     for(int i=0; i<length; ++i)
16     {
17         if(mx>i)
18         {
19             p[i] = min(p[2*id-i], mx-i);
20         }
21         else
22         {
23             p[i] = 1;
24         }
25 
26         while( (i-p[i]+1)>=0 && (i+p[i]-1)<length && cArray[i-p[i]+1]==cArray[i+p[i]-1] )
27         {
28             p[i] = p[i] + 1;
29         }
30 
31         p[i]--;
32 
33         if(i+p[i]-1 > mx)
34         {
35             mx = i+p[i]-1;
36             id = i;
37         }
38 
39         if(maxLength < p[i]-1)
40         {
41             maxLength = p[i]-1;
42         }
43     }
44 
45 
46     return maxLength;
47 }
48 
49 
50 
51 
52 int main()
53 {
54     //string input = "waabwswfd";
55     string input = "wawbbbwasaw";
56     int k = 0;
57     for(int i=0; i<input.size(); ++i)
58     {
59         cArray[k++] = '#';
60         cArray[k++] = input.at(i);
61     }
62     cArray[k++] = '#';
63     int ans = manacher(k);
64     cout << ans << endl;
65 }

 

 简化以后的代码

#include<iostream>
#include<string>
#include<cstdlib>
#include<algorithm>

using namespace std;

char cArray[1000];
int p[1000];

int manacher(int length)
{
    int mx = 0;
    int id = 0;
    int maxLength = 0;

    for (int i = 0; i<length; ++i)
    {
        if (mx>i)
        {
            p[i] = min(p[2 * id - i], mx - i);
        }
        else
        {
            p[i] = 1;
        }

        while ((i - p[i]) >= 0 && (i + p[i])<length && cArray[i - p[i]] == cArray[i + p[i]])
        {
            p[i] = p[i] + 1;
        }

        p[i]--;

        if (i + p[i] > mx)
        {
            mx = i + p[i];
            id = i;
        }

        if (maxLength < p[i])
        {
            maxLength = p[i];
        }
    }


    return maxLength;
}




int main()
{
    //string input = "waabwswfd";
    string input = "wawbbbwasaw";
    int k = 0;
    for (int i = 0; i<input.size(); ++i)
    {
        cArray[k++] = '#';
        cArray[k++] = input.at(i);
    }
    cArray[k++] = '#';
    int ans = manacher(k);
    cout << ans << endl;
}

 

 

 

 

posted on 2015-04-01 13:24  午夜稻草人  阅读(244)  评论(0编辑  收藏  举报