[HAOI2008]玩具取名（区间 dp）

玩具取名

题目描述

某人有一套玩具，并想法给玩具命名。首先他选择WING四个字母中的任意一个字母作为玩具的基本名字。然后他会根据自己的喜好，将名字中任意一个字母用“WING”中任意两个字母代替，使得自己的名字能够扩充得很长。
现在，他想请你猜猜某一个很长的名字，最初可能是由哪几个字母变形过来的。

输入描述:

第一行四个整数W、I、N、G。表示每一个字母能由几种两个字母所替代。
接下来W行，每行两个字母,表示W可以用这两个字母替代。
接下来I行，每行两个字母,表示I可以用这两个字母替代。
接下来N行，每行两个字母,表示N可以用这两个字母替代。
接下来G行，每行两个字母,表示G可以用这两个字母替代。
最后一行一个长度不超过Len的字符串。表示这个玩具的名字。

输出描述:

一行字符串，该名字可能由哪些字母变形而得到。（按照WING的顺序输出）
如果给的名字不能由任何一个字母变形而得到则输出“The name is wrong!”

输入

1 1 1 1
II
WW
WW
IG
IIII

输出

IN

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <map>

using namespace std;
map<char, int> m;
int f[5][20][3];
int num[5];
char use[2];
char s[210];
char ans[6] = {"0WING"};
bool dp[210][210][5];

int main() {
    m['W'] = 1, m['I'] = 2, m['N'] = 3, m['G'] = 4;
    for (int i = 1; i <= 4; i++) {
        scanf("%d", &num[i]);
    }
    for (int i = 1; i <= 4; i++) {
        for (int j = 1; j <= num[i]; j++) {
            scanf("%s", use);
            f[i][j][1] = m[use[0]];
            f[i][j][2] = m[use[1]];
        }
    }
    scanf("%s", s + 1);
    int len = strlen(s + 1);
    for (int i = 1; i <= len; i++) {
        dp[i][i][m[s[i]]] = 1;
    }
    //枚举区间长度
    for (int i = 2; i <= len; i++) {
        //枚举区间左端点和右端点
        for (int l = 1; l + i <= len + 1; l++) {
            int r = l + i - 1;
            //两区间交界处
            for (int j = 1; j < i; j++) {
                //枚举 WING
                for (int k = 1; k <= 4; k++) {
                    //WING 能被组合出的方案
                    for (int z = 1; z <= num[k]; z++) {
                        //标记
                        if (dp[l][l + j - 1][f[k][z][1]] && dp[l + j][r][f[k][z][2]]) {
                            dp[l][r][k] = 1;
                        }
                    }
                }
            }
        }
    }
    int flag = 0;
    for (int i = 1; i <= 4; i++) {
        if (dp[1][len][i]) {
            flag = 1;
            printf("%c", ans[i]);
        }
    }
    if (!flag) {
        printf("The name is wrong!\n");
    }
    return 0;
}