Luogu P5446 [THUPC2018]绿绿和串串
Link
首先显然合法的\(T\)都是\(S\)的\(pre\)。
如果一个位置的回文半径达到了\(S\)的末尾,那么这个位置一定是合法的。
同时如果一个位置的回文半径达到了\(S\)的开头并且末尾的位置是合法的,那么这个位置也是合法的。
可以证明其它的位置一定不合法。
#include<cstdio>
#include<cstring>
const int N=1000006;
char s[N];int len[N],f[N];
int read(){int x;scanf("%d",&x);return x;}
int min(int a,int b){return a<b? a:b;}
int main()
{
for(int T=read();T;--T)
{
scanf("%s",s+1),memset(len,0,sizeof len),memset(f,0,sizeof f);
int n=strlen(s+1);s[n+1]='$';
for(int i=1,pos=0,mx=0;i<=n;++i)
{
len[i]=i<mx? min(len[pos*2-i],mx-i):1;
while(s[i-len[i]]==s[i+len[i]]) ++len[i];
if(mx<i+len[i]) pos=i,mx=i+len[i];
}
for(int i=n;i;--i) if(i+len[i]-1==n||(i==len[i]&&f[i+len[i]-1])) f[i]=1;
for(int i=1;i<=n;++i) if(f[i]) printf("%d ",i);
puts("");
}
}