百度笔试题--最长回文字串

#include <iostream>
using namespace std;

void substring(char* str)
{    
    int len = strlen(str);
    len = 2*len + 1;
    for(int i = len; i >= 1; i--)
    {
        if(i%2 == 0)
            str[i-1] = str[i/2-1];
        else
            str[i-1] = '#';
    }
    str[len] = '';
    int* p = new int[len]; //p[i]用来保存第i个字符为中心的回文字串向右/向左扩张的距离
    int mx = 0; //表示已经处理的回文字串向右扩张的最长距离
    int id = 0; //表示上述回文子串的中间位置
    memset(p,0,len);
    for(int i = 0; i < len; i++)
    {
        if(mx - i > 0)
        {
            p[i] = p[2*id - i] < (mx - i)?p[2*id - i]:(mx-i);
        }
        else
            p[i] = 1;
        while(str[i + p[i]] == str[i - p[i]]) p[i]++;
        if(p[i] > mx - i)
        {
            mx = i + p[i];
            id = i;
        }
    }
    mx = 0;
    id = 0;
    for(int i = 0; i < len; i++)
    {
        if(p[i] > mx)
        {
            mx = p[i];
            id = i;
        }
    }
    cout<<"最长回文字串：";
    for(int i = id-mx+1; i < id+mx; i++)
    {
        if(str[i] != '#')
            cout<<str[i];
    }
    cout<<endl;
}
int main()
{
    while(1)
    {
        char* str = new char[100];
        if(str == NULL)
            return 0;
        cout<<"输入字符串：";
        cin>>str;
        substring(str);
        delete[] str;
        str = NULL;
    }
    return 0;
}