[BZOJ 4060] Word Equations

Link:

BZOJ 4060 传送门

Solution:

可以发现字符串间的关系可以构成一棵树

于是我们先用字符串哈希建树，再树形$dp$即可

设$dp[i][j]$为第$i$个节点从$P$字符串的第$j$为开始匹配的失配位置，

则有$dp[i][j]=dp[ch[i][1]][dp[ch[i][0]][j]]$

在叶子节点时暴力匹配就行了，复杂度$O(k*P)$

Code:

#include <bits/stdc++.h>

using namespace std;

const int MAXN=,ML=;
int T,n,hs[MAXN],dp[MAXN][ML],ch[MAXN][],root;
char S[ML],P[ML],dat[MAXN][ML],tmp[ML],a[ML];

inline int hash(char t[])
{
    int ret=;
    for(int i=;i<;i++)
        ret=ret*+((i<strlen(t))?(t[i]-'A'+):);
    return ret;
}

int dfs(int x,int y)
{
    if(dp[x][y]!=-) return dp[x][y];
    if(ch[x][]) return dp[x][y]=dfs(ch[x][],dfs(ch[x][],y));
    int k=y;
    for(int cur=;k<strlen(P) && cur<strlen(dat[x]);cur++)
        if(dat[x][cur]==P[k]) k++;
    return dp[x][y]=k;
}

int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);memset(dp,-,sizeof(dp));
        for(int i=;i<=n;i++)
        {
            scanf("%s%s%s",a,tmp,dat[i]);hs[i]=hash(a);
            if(dat[i][]>='a' && dat[i][]<='z') ch[i][]=ch[i][]=;
            else
            {
                ch[i][]=hash(dat[i]);
                scanf("%s%s",tmp,dat[i]);
                ch[i][]=hash(dat[i]);
            }
        }

        for(int i=;i<=n;i++) if(ch[i][])
            for(int j=;j<;j++) for(int k=;k<=n;k++)
                if(ch[i][j]==hs[k]) ch[i][j]=k;
        scanf("%s%s",S,P);
        for(int i=;i<=n;i++) if(hs[i]==hash(S)) root=i;

        printf((dfs(root,)<strlen(P))?"NO\n":"YES\n");
    }
    return ;
}

Review:

如要对字符数组大量比较+判等于，最好使用字符串$Hash$ （采用27进制）

如果不考虑$string$输入效率较低，则对$string$用$map$或$hash\_ map$更优