LightOJ 1051 Good or Bad (递推DP)

题意

如果一个字符串含有连续的3个元音或者连续的5个辅音那么他就是bad的，否则他就是good的，字符串中含有’?’，’?’有可能是任何字母，如果因为问号不确定的话就是mix。

思路

因为数据很小所以刚开始直接考虑模拟，但是问号的处理很麻烦，换成了dp的做法。

dp的方程还是比较显然的，但是同样是mixed不太好判断。

dp[0/1][i][j]表示在i前第j个位时元音和辅音的情况，最后判断good即可。

代码

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define LL long long
#define Lowbit(x) ((x)&(-x))
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1|1
#define MP(a, b) make_pair(a, b)
const int INF = 0x3f3f3f3f;
const int MOD = 1000000007;
const int maxn = 1000 + 7;
const double eps = 1e-8;
const double PI = acos(-1.0);
typedef pair<int, int> pii;

char s[55];
int dp[2][55][10];

bool vow(char ch)
{
return (ch == 'A' || ch == 'E' || ch == 'I' || ch == 'O' || ch == 'U');
}

int solve()
{
int len = strlen(s + 1);
memset(dp, 0, sizeof(dp));
dp[0][0][0] = dp[1][0][0] = 1;
bool good = 0, bad = 0;
for (int i = 1; i <= len; i++)
{
if (vow(s[i]) || s[i] == '?')
{
for (int j = 0; j < 2; j++)
{
dp[0][i][j+1] |= dp[0][i-1][j];
dp[1][i][0] |= dp[0][i-1][j];
}
if (dp[0][i-1][2]) bad = 1;
}
if (!vow(s[i]) || s[i] == '?')
{
for (int j = 0; j < 4; j++)
{
dp[1][i][j+1] |= dp[1][i-1][j];
dp[0][i][0] |= dp[1][i-1][j];
}
if (dp[1][i-1][4]) bad = 1;
}
}
for (int i = 0; i < 6; i++)
good |= (dp[0][len][i] | dp[1][len][i]);
if (good && bad) return 3;
if (good && !bad) return 1;
if (!good && bad) return 2;
}

int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int T;
scanf("%d", &T);
for (int ncase = 1; ncase <= T; ncase++)
{
scanf("%s", s + 1);
int flag = solve();  //1good 2bad 3mix
printf("Case %d: ", ncase);
if (flag == 1) puts("GOOD");
else if (flag == 2) puts("BAD");
else if (flag == 3) puts("MIXED");
}
return 0;
}