Russian Code Cup 2017 - Finals B. Similar Words(hash+树形dp)

题目链接：Russian Code Cup 2017 - Finals  B. Similar Words

题意：

给你n个字符串，现在让你构造一个集合X。

1.使得X的每个元素都是这n个字符串中的前缀。

2.X中任意两个元素都不相似(即一个字符串去掉第一个字符后不是另一个字符串)

题解：

我们将所有的前缀取出来，考虑将相似的连边。这里可以发现一个字符串去掉首字符后只有唯一的一个字符串。

所有将相似的连边后，就形成了一个森林。(判相似可以用hash，可以用AC自动机)

然后建完边后就是一个树形dp了。

这里hash容易被卡，双hash1.5s飘过。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
#define mst(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long uu;
typedef pair<pair<uu,uu>,int>P;
namespace str_hash
{
    const uu sd=1e9+7,D=1e9+9,N=1e6+7;
    uu b[N],B[N];
    void init(){
        b[0]=1;F(i,1,N-1)b[i]=(b[i-1]*31)%sd;
        B[0]=1;F(i,1,N-1)B[i]=(B[i-1]*31)&((1<<30)-1);
    }
    struct STR{
        int l;
        uu *h,*h2;
        void ins(char *s){
            int len=strlen(s+1);
            h=new uu[len+1];
            h2=new uu[len+1];
            for(h[0]=0,h2[0]=0,l=1;s[l];l++)
            {
                h[l]=(h[l-1]*31+s[l])%sd;
                h2[l]=(h2[l-1]*31+s[l])&((1<<30)-1);
            }
            l--;
        }
        pair<uu,uu> ask(int l,int r){
            return pair<uu,uu>((h[r]-h[l-1]*b[r-l+1]%sd+sd)%sd,(h2[r]-(h2[l-1]*B[r-l+1]&((1<<30)-1))+((1<<30)-1))&((1<<30)-1));
        }
    }str[N];
}

using namespace str_hash;
int t,n;
char s[N];
vector<int>g[N];
set<P>st;
int dp[N][2],in[N],vis[N];

void dfs(int x)
{
    dp[x][0]=1,dp[x][1]=0;
    for(auto &it:g[x])
    {
        dfs(it);
        dp[x][0]+=dp[it][1];
        dp[x][1]+=max(dp[it][0],dp[it][1]);
    }
}


int main(){
    init(),scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        F(i,1,n)
        {
            scanf("%s",s+1);
            str[i].ins(s);
        }
        int idx=0;
        F(i,1,n)F(j,1,str[i].l)
        {
            ++idx;
            pair<uu,uu> now=str[i].ask(1,j);
            auto it=st.lower_bound(P(now,0));
            if(it!=st.end()&&it->first==now)vis[idx]=1;
            else st.insert(P(now,idx));
        }
        idx=0;
        F(i,1,n)F(j,1,str[i].l)
        {
            ++idx;
            if(j==1||vis[idx])continue;
            pair<uu,uu> now=str[i].ask(2,j);
            auto it=st.lower_bound(P(now,0));
            if(it!=st.end()&&it->first==now)
            {
                g[it->second].push_back(idx);
                in[idx]++;
            }
        }
        int ans=0;
        F(i,1,idx)if(in[i]==0&&vis[i]==0)
        {
            dfs(i),ans+=max(dp[i][0],dp[i][1]);
        }
        printf("%d\n",ans);
        st.clear();
        F(i,1,idx)dp[i][0]=dp[i][1]=0,g[i].clear(),vis[i]=in[i]=0;
    }
    return 0;
}