D - We Like AGC 动态规划

题目链接 && 题意

D - We Like AGC
给定数字n，请统计有多少长度为n，且只包括'A','C','G','T'四个字符的字符串个数。并且这些字符串要满足如下两个条件:

1. 没有子串"AGC"
2. 交换相邻两位，也不会出现子串"AGC"

思路

难点就在于如何避免子串出现"AGC"!!
若$dp[l][i][j][k]$表示长度为l且最后三个字符为i,j,k且满足条件的字符串数量，注意此处i,j,k只是代表字符，可以定义一个字符数组：

const char ch[] = {'A', 'C', 'G', 'T'};

可以用来表示数组的下标，就能分别对应为字符了。
那么转移方程就是$dp[l+1][i][j][k]=dp[l][0][i][j]+dp[l][1][i][j]+dp[l][2][i][j]+dp[l][3][i][j]$，这个转移方程大致是对的，为什么说大致是对的，因为新生成的字符串可能又不能满足条件了，故需要先判断一下。如果新的字符为'k'，前三个字符为'u','i','j'，那么新生成的字符串最后四位为'u','i','j','k'，应该判断这四个字符组成的子串是否满足条件，如果满足条件才相加。

AC代码

#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <string>
using namespace std;
#define MOD 1000000007;
const int maxn = 100 + 5;
const char ch[] = {'A', 'C', 'G', 'T'};

int dp[maxn][4][4][4]; 

bool isValid(int x, int y, int z) {
    string str = "";
    str.push_back(ch[x]);
    str.push_back(ch[y]);
    str.push_back(ch[z]);
    if(str == "AGC")
        return false;
    return true;
}

int solve(int n) {
    memset(dp, 0, sizeof(dp));
    for(int i = 0; i < 4; i++)
        for(int j = 0; j < 4; j++) {
            for(int k = 0; k < 4; k++) {
                if(isValid(i, j, k) && isValid(j, i, k) && isValid(i, k, j)) {
                    dp[3][i][j][k] = 1;
                }
            }
        }

    for(int l = 4; l <= n; l++) {
        for(int i = 0; i < 4; i++) {
            for(int j = 0; j < 4; j++) {
                for(int k = 0; k < 4; k++) {
                    for(int u = 0; u < 4; u++) {
                        if(isValid(i, j, k) && isValid(u, i, k) && isValid(u, j, k) &&  isValid(j, i, k) && isValid(i, k, j)) {
                            dp[l][i][j][k] += dp[l-1][u][i][j] % MOD;
                            dp[l][i][j][k] %= MOD;
                        }
                    }
                }
            }
        }
    }

    int res = 0;
    for(int i = 0; i < 4; i++) {
        for(int j = 0; j < 4; j++) {
            for(int k = 0; k < 4; k++) {
                res += dp[n][i][j][k];
                res %= MOD;
            }
        }
    }
    return res;
}


int main() {
#ifdef LOCAL
    freopen("data.in", "r", stdin);
#endif // LOCAL
    int n;
    while(scanf("%d", &n) == 1) {
        printf("%d\n", solve(n));
    }
    return 0;
}

如有不当之处欢迎指出.