AtCoder Beginner Contest 122 D - We Like AGC(DP)

题目链接

思路自西瓜and大佬博客：https://www.cnblogs.com/henry-1202/p/10590327.html#_label3

数据范围小 可直接dp

f[i][j][a][b] 表示 i位置上是j i-1上是a i-2上是b

状态转移是由i-1转移过来的，所以就必须还要一个i-3 所以就多加上一个循环

最主要就是转移过程中要枚举每种情况 然后排除掉

大佬的博客用了map和string简化了枚举的过程

不过他说只有六种情况 我布吉岛为啥只有六种 我写出了八种qaq

map<int, string> mp;
bool check(int a, int b, int c, int d) {
    if (mp[a] + mp[b] + mp[c] == "AGC") return ;
    if (mp[a] + mp[c] + mp[b] == "AGC") return ;
    if (mp[a] + mp[b] + mp[d] == "AGC") return ;
    if (mp[a] + mp[c] + mp[d] == "AGC") return ;
    if (mp[b] + mp[a] + mp[c] == "AGC") return ;
    if (mp[b] + mp[c] + mp[d] == "AGC") return ;
    if (mp[b] + mp[d] + mp[c] == "AGC") return ;
    if (mp[c] + mp[b] + mp[d] == "AGC") return ;
    return ;
}

mp[0]='E'就是来处理刚开始未到3个字符时候的情况 不过枚举j也就是第i个位置的字符就从1开始了 所以之后的不会受到mp[0]的影响了

#include <cstdio>
#include <algorithm>
#include <string>
#include <map>
using namespace std;

map<int, string> mp;
long long f[][][][];
const long long mod = 1e9 + ;

bool check(int a, int b, int c, int d) {
    if (mp[a] + mp[b] + mp[c] == "AGC") return ;
    if (mp[a] + mp[c] + mp[b] == "AGC") return ;
    if (mp[a] + mp[b] + mp[d] == "AGC") return ;
    if (mp[a] + mp[c] + mp[d] == "AGC") return ;
    if (mp[b] + mp[a] + mp[c] == "AGC") return ;
    if (mp[b] + mp[c] + mp[d] == "AGC") return ;
    if (mp[b] + mp[d] + mp[c] == "AGC") return ;
    if (mp[c] + mp[b] + mp[d] == "AGC") return ;
    return ;
}

int main() {
    int n;
    scanf("%d", &n);
    f[][][][] = ;
    mp[] = 'E', mp[] = 'A', mp[] = 'C', mp[] = 'G', mp[] = 'T';
    for (int i = ; i <= n; i++)
        for (int j = ; j <= ; j++)
            for (int a = ; a <= ; a++)
                for (int b = ; b <= ; b++)
                    for (int l = ; l <= ; l++)
                        if (check(l, b, a, j))
                            f[i][j][a][b] = (f[i][j][a][b] + f[i-][a][b][l]) % mod;
    long long ans = ;
    for (int i = ; i <= ; i++)
        for (int j = ; j <= ; j++)
            for (int k = ; k <= ; k++)
                ans = (ans + f[n][i][j][k]) % mod;
    printf("%lld\n", ans);
    return ;
}