BZOJ 1055 [HAOI2008]玩具取名

题目链接

一个人首先给一个玩具取名为WING中的一个字符,且W,I,N,G分别可以变成w,i,n,g种两个WING中的字符,给出玩具最终的名字,问最初的名字可以为那些字符

定义dp[l][r][k]为[l,r]是否可以变成一种第k个字符

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
int wing[4]; char WING[4][20][5];
char s[205];
int f[205][205][4];
int arc(char c) {
	switch(c) {
		case 'W': return 0;
		case 'I': return 1;
		case 'N': return 2;
		case 'G': return 3;
	}
}
int DP(int l, int r, int k) {
	if(l == r) return arc(s[l]) == k;
	int & res = f[l][r][k];
	if(res != -1) return res;
	res = false;
	for(int i = l; i < r; i++) {
		for(int j = 0; j < wing[k]; j++)
		  res |= DP(l, i, arc(WING[k][j][0])) 
			  && DP(i + 1, r, arc(WING[k][j][1]));
	}
	return res;
}
int main() {
	for(int i = 0; i < 4; i++) scanf("%d", &wing[i]);
	for(int i = 0; i < 4; i++) 
	  for(int j = 0; j < wing[i]; j++) 
		scanf("%s", WING[i][j]);
	scanf("%s", s);
	int len = strlen(s);
	memset(f, -1, sizeof(f));
	bool ans = 1;
	if(DP(0, len - 1, 0)) putchar('W'), ans = 0;
	if(DP(0, len - 1, 1)) putchar('I'), ans = 0;
	if(DP(0, len - 1, 2)) putchar('N'), ans = 0;
	if(DP(0, len - 1, 3)) putchar('G'), ans = 0;
	if(ans) printf("The name is wrong!");
	putchar('
');
	return 0;
}