携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第14天，点击查看活动详情

[HAOI2008] 玩具取名

题目描述

某人有一套玩具，并想法给玩具命名。首先他选择WING四个字母中的任意一个字母作为玩具的基本名字。然后他会根据自己的喜好，将名字中任意一个字母用“WING”中任意两个字母代替，使得自己的名字能够扩充得很长。

现在，他想请你猜猜某一个很长的名字，最初可能是由哪几个字母变形过来的。

输入格式

第一行四个整数W、I、N、G。表示每一个字母能由几种两个字母所替代。

接下来W行，每行两个字母,表示W可以用这两个字母替代。

接下来I行，每行两个字母,表示I可以用这两个字母替代。

接下来N行，每行两个字母,表示N可以用这两个字母替代。

接下来G行，每行两个字母,表示G可以用这两个字母替代。

最后一行一个长度不超过Len的字符串。表示这个玩具的名字。

输出格式

一行字符串，该名字可能由哪些字母变形而得到。（按照WING的顺序输出）

如果给的名字不能由任何一个字母变形而得到则输出“The name is wrong!”

样例 #1

样例输入 #1

1 1 1 1
II
WW
WW
IG
IIII

样例输出 #1

IN

提示

30%数据满足Len<=20，W、I、N、G<=6

100%数据满足Len<=200，W、I、N、G<=16

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

char sss[201];
int m[618], s[5];
bool ok[5][5][5], f[201][201][5], flag;

int main() {
    m['W'] = 1, m['I'] = 2, m['N'] = 3, m['G'] = 4; 
    for (int i = 1; i <= 4; ++i) scanf("%d", &s[i]);
    for (int i = 1; i <= 4; ++i) {
        for (int j = 1; j <= s[i]; ++j) {
            char a, b;
            std::cin >> a >> b;
            ok[m[a]][m[b]][i] = 1;
        }
    }
    std::cin >> sss;
    int n = strlen(sss);
    for (int i = 1; i <= n; ++i) {
        f[i][i][m[sss[i - 1]]] = 1;
    }
    for (int l = 2; l <= n; ++l) {
        for (int i = 1; i <= n - l + 1; ++i) {
            int j = i + l - 1;
            for (int k = i; k < j; ++k) {
                for (int x = 1; x <= 4; ++x) {
                    for (int y = 1; y <= 4; ++y) {
                        for (int z = 1; z <= 4; ++z) {
                            if (f[i][k][y] && f[k + 1][j][z] && ok[y][z][x]) {
                                f[i][j][x] = 1;
                            }
                        }
                    }
                }
            }
        }
    }
    if (f[1][n][1]) flag = 1, printf("%c", 'W');
    if (f[1][n][2]) flag = 1, printf("%c", 'I');
    if (f[1][n][3]) flag = 1, printf("%c", 'N');
    if (f[1][n][4]) flag = 1, printf("%c", 'G');
    if (!flag) puts("The name is wrong!");
    return 0;
}