codeforces D. Palindrome pairs 动态规划

刚开始作道题 我只是想到一个 n四次方的算法  不敢打啊!!!后来和磊哥讨论讨论 得到一个 n二次方的算法 还可以 就是

计算出一i为尾节点的回文个数dp1[i],计算出以i为头节点的个数dp2[i],这样只要做相乘就可以啦 这个事n方,

之前处理dp1[i],dp2[i]也是n方的算法 详细看代码吧!!!!

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
char a[2005];
long long dps[2005],dpe[2005];
int main()
{
    while(~scanf("%s",&a))
    {
        int len=strlen(a);
        for(int i=0;i<=len;i++)
        {
            dps[i]=dpe[i]=1;
        }
        for(int i=0;i<len;i++)//偶数回文 
        {
            for(int j=0;j<(len/2+1);j++)
            {
                if((i-j)<0||(i+j+1)>=len)
                break;
                if(a[i-j]==a[i+1+j])
                {
                    dps[i-j]++;
                    dpe[i+1+j]++;
                }
                else
                break;
            }
        }
        for(int i=1;i<len;i++)  //奇数回文 
        {
            for(int j=1;j<(len/2+1);j++)
            {
                if((i-j)<0||(i+j)>=len)
                break;
                if(a[i-j]==a[i+j])
                {
                    dps[i-j]++;
                    dpe[i+j]++;
                }
                else
                break;
            }
        }
        long long ans=0;
        for(int i=0;i<len;i++)
        {
            long long sum=0;
            for(int j=i+1;j<len;j++)
            sum+=dps[j];
            ans+=sum*dpe[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}


 

你可能感兴趣的:(codeforces D. Palindrome pairs 动态规划)