济南学习 Day5 T3 晚

回文串(palindromes)

【题目描述】

判断是否能将字符串S分成三段非空回文串。

【输入说明】

第一行一个整数T，表示数据组数。

对于每一个组，仅包含一个由小写字母组成的串。

【输出说明】

对于每一组，单行输出"Yes" 或 "No"。

【样例输入】

2

abc

abaadada

【样例输出】

Yes

No

【数据范围】

对于40%的数据，|S|<=100

对于60%的数据，|S|<=1000

对于100%的数据，T<=20，|S|<=20000

#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 20010
#define P 29
#define ll long long 
using namespace std;
ll T,n,A[maxn],B[maxn],a[maxn],b[maxn],p[maxn],flag;
char s[maxn];
void Get()
{
    p[0]=1;
    for(int i=1;i<=20000;i++)
      p[i]=p[i-1]*P;
}
void Hash1()
{
    A[0]=0;
    for(int i=1;i<=n;i++)
      A[i]=A[i-1]*P+s[i];
}
void Hash2()
{
    B[0]=0;
    for(int i=1;i<=n;i++)
      B[i]=B[i-1]*P+s[n-i+1];
}
ll query2(ll l,ll r)
{
    return B[r]-B[l-1]*p[r-l+1];
}
ll query1(ll l,ll r)
{
    return A[r]-A[l-1]*p[r-l+1];
}
int main()
{
    scanf("%d",&T);
    Get();
    while(T--)
    {
        scanf("%s",s+1);
        n=strlen(s+1);
        a[0]=0;b[0]=0;
        Hash1();Hash2();flag=0;
        for(int i=1;i<=n;i++)
        {
            ll x=query1(1,i);
            ll y=query2(n-i+1,n);
            if(x==y) a[++a[0]]=i;
        }
        for(int i=n;i>=1;i--)
        {
            ll x=query1(i,n);
            ll y=query2(1,n-i+1);
            if(x==y) b[++b[0]]=i;
        }
        for(int i=1;i<=a[0];i++)
        {
            for(int j=1;j<=b[0];j++)
            {
                if(a[i]+1>b[j]-1) break;
                ll x=query1(a[i]+1,b[j]-1);
                ll y=query2(n-b[j]+2,n-a[i]);
                if(x==y)
                {
                    flag=1;break;
                }
            }
            if(flag) break;
        }    
        if(flag) printf("Yes
");
        else printf("NO
");
    }
    
    
    return 0;
}

八十分做法~~~，已跪，求大神解救~~