bzoj1055: [HAOI2008]玩具取名 dp

dp[l][r][a]代表区间l到r是否能变为a

然后直接记忆化搜索。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <cmath>
#include <algorithm>
using namespace std;
char p[4]={'W','I','N','G'};
int q[255],t[4];;
char ch[205],a[4][20][2];
int dp[205][205][4];
int dfs(int l,int r,int num)
{
    if(dp[l][r][num]!=-1) return dp[l][r][num];
    if(l==r) return dp[l][r][num]=(p[num]==ch[l]);
 
    int tmp=0;
    for(int i=1;i<=t[num];i++)
    {
        for(int j=l;j<r;j++)
        {
            if(dfs(l,j,q[a[num][i][0]])&&dfs(j+1,r,q[a[num][i][1]]))
            {
                tmp=1;
                break;
            }
        }
    }
    return dp[l][r][num]=tmp;
}
int main()
{
    memset(dp,-1,sizeof(dp));
    q['W']=0;q['I']=1;q['N']=2;q['G']=3;
    for(int i=0;i<4;i++)
        scanf("%d",&t[i]);
    for(int i=0;i<4;i++)
        for(int j=1;j<=t[i];j++)
            scanf("%s",a[i][j]);
    scanf("%s",ch+1);
    int n=strlen(ch+1);
    bool flag=0;
    for(int i=0;i<4;i++)
        if(dfs(1,n,i))
        {
            flag=1;
            printf("%c",p[i]);
        }
    if(!flag)
    {
        puts("The name is wrong!");
    }
    return 0;
}