Codeforces 653B Bear and Compressing 【数位dp】

题目链接：Codeforces 653B Bear and Compressing

题意：给定q个a[i]−b[i]串，若串s开头的两个字符 ==a[i]，那么我们可以把它们消去换成一个b[i]。最后得到字符a，问s串的方案数。

思路：数位dp嘛。不过暴力也是可以过得。

dp[i][j][flag]表示处理到第i位，前一位填j，flag表示状态是否合法。

AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#define PI acos(-1.0)
#define CLR(a, b) memset(a, (b), sizeof(a))
#define fi first
#define se second
#define ll o<<1
#define rr o<<1|1
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
const int MAXN = 1e5 + 10;
const int pN = 1e6;// <= 10^7
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
void add(LL &x, LL y) { x += y; x %= MOD; }
string a[40], b[40];
int dp[10][10][2];
int n, q;
int DFS(int pos, int pre, bool flag) {
if(pos == 0) {
return flag;
}
if(dp[pos][pre][flag] != -1) return dp[pos][pre][flag];
int ans = 0;
for(int i = 0; i < q; i++) {
ans += DFS(pos-1, a[i][0] - 'a', flag&&(b[i][0] - 'a' == pre));
}
dp[pos][pre][flag] = ans;
return ans;
}
int main()
{
cin >> n >> q;
for(int i = 0; i < q; i++) {
cin >> a[i] >> b[i];
}
CLR(dp, -1);
cout << DFS(n-1, 0, 1) << endl;
return 0;
}