hdu5469 Antonidas 2015上海网络赛

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;
}