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;
}