http://acm.hdu.edu.cn/showproblem.php?pid=5469
点分治加字符串hash
先预处理出要找到的字符串的前缀和后缀的hash值,使用ull自然溢出,幂取131(之前听说字符串hash这样比较好
那么我们再点分治的时候,知道从当前rt向下到每个点的深度dis[j],也知道它的hash值a[j],如果刚好和pre[dis[j]]或者suf[slen-dis[j]+1]相等的话,说明这可以是一条前缀或者后缀,那么只要找对应的后缀或前缀是否已经出现就行了
(一开始用map存前缀对应的后缀,和后缀对应的前缀,多了个log超时了
注意n=1的情况,WA了一发
#include<bits/stdc++.h>
#define ull unsigned long long
using namespace std;
const int maxl=2e4+10;
int n,m,top1,top2,len,cnt,cas,slen,ans;
int ehead[maxl],dis[maxl],s1[maxl],s2[maxl];
ull pre[maxl],suf[maxl],mi[maxl],a[maxl];
bool numpre[maxl],numsuf[maxl];
char c[maxl],s[maxl];
bool vis[maxl];
struct ed
{
int to,nxt;
}e[maxl<<1];
struct centertree
{
int ans,n,mi;
int son[maxl];
inline void dfs(int u,int fa)
{
son[u]=1;int v,mx=0;
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(vis[v] || v==fa) continue;
dfs(v,u);
son[u]+=son[v];
mx=max(son[v],mx);
}
mx=max(mx,n-son[u]);
if(mx<mi)
mi=mx,ans=u;
}
inline int getcenter(int u)
{
ans=0;mi=maxl;
dfs(u,0);
return ans;
}
}tree;
inline void add(int u,int v)
{
e[++cnt].to=v;e[cnt].nxt=ehead[u];ehead[u]=cnt;
}
inline void prework()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
ehead[i]=0,vis[i]=false;
int u,v;cnt=0;
for(int i=1;i<n;i++)
{
scanf("%d%d",&u,&v);
add(u,v);add(v,u);
}
scanf("%s%s",c+1,s+1);
slen=strlen(s+1);suf[slen+1]=0;
for(int i=1;i<=slen;i++)
{
pre[i]=pre[i-1]*131+s[i]-'a'+1;
suf[slen-i+1]=suf[slen-i+2]*131+s[slen-i+1]-'a'+1;
}
}
inline void dfs(int u,int fa,ull w,int dep)
{
w=w+mi[dep]*(c[u]-'a'+1);
a[++len]=w;dis[len]=dep+1;
int v;
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(vis[v] || v==fa) continue;
dfs(v,u,w,dep+1);
}
}
inline void solv(int u)
{
vis[u]=true;
int v,rt;top1=0;top2=0;
if(c[u]-'a'+1==pre[1])
{
numpre[1]=true;
s1[++top1]=1;
}
if(c[u]-'a'+1==suf[slen])
{
if(numpre[slen])
ans=1;
numsuf[slen]=true;
s2[++top2]=slen;
}
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(vis[v]) continue;
len=0;
dfs(v,u,c[u]-'a'+1,1);
for(int j=1;j<=len;j++)
{
if(a[j]==pre[dis[j]])
{
if(numsuf[dis[j]])
ans=1;
}
if(a[j]==suf[slen-dis[j]+1])
{
if(numpre[slen-dis[j]+1])
ans=1;
}
}
for(int j=1;j<=len;j++)
{
if(a[j]==pre[dis[j]])
numpre[dis[j]]=true,s1[++top1]=dis[j];
if(a[j]==suf[slen-dis[j]+1])
numsuf[slen-dis[j]+1]=true,s2[++top2]=slen-dis[j]+1;
}
}
for(int i=1;i<=top1;i++)
numpre[s1[i]]=false;
for(int i=1;i<=top2;i++)
numsuf[s2[i]]=false;
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(vis[v]) continue;
tree.n=tree.son[v];
rt=tree.getcenter(v);
solv(rt);
}
}
inline void mainwork()
{
ans=0;tree.n=n;
int rt=tree.getcenter(1);
solv(rt);
}
inline void print()
{
printf("Case #%d: ",cas);
if(ans)
puts("Find");
else
puts("Impossible");
}
int main()
{
mi[0]=1;
for(int i=1;i<maxl;i++)
mi[i]=mi[i-1]*131;
int t;
scanf("%d",&t);
for(cas=1;cas<=t;cas++)
{
prework();
mainwork();
print();
}
return 0;
}