POJ Apocalypse Someday

POJ3208

还是教练强啊,这个写法可以推广到很广哎qwq

Code：

#include <cstdio>
using namespace std;
int T, n, m;
int f[21][4];//f[i][j]表示从j状态(当前有几个连续的6)走i步到达目标状态(有连续3个6)所有方案数 
int g[4][21];//g[i][j]表示当前已有i个连续的6,在后面添加j后到达哪个状态 
int main () {
    for (int i = 0; i < 3; i++) {
        g[i][6] = i + 1;//后面添加6就会相应的到下一个状态 
    }
    for (int i = 0; i <= 9; i++) {
        g[3][i] = 3;//当前已经有3个连续的6之后,后面加什么都会回到当前状态 
    }
    f[0][3] = 1;//初始化 
    for (int i = 1; i <= 10; i++) {//枚举所有步数,也就是多少位 
        for (int j = 0; j <= 3; j++) {//连续几个6 
            for (int k = 0; k <= 9; k++) {//添加哪个数字 
                f[i][j] += f[i - 1][g[j][k]];//状态转移 
            }
        }
    }
    scanf("%d", &T);
    while (T--) {
        scanf("%d", &n);
        for (m = 1; m < 11 && f[m][0] < n; m++) {
            //确定目标状态的位数 
        }
        int p = 0;
        for (int i = m, j = 0; i; i--) {//倒叙循环枚举到了哪一位 
            for (j = 0; j <= 9; j++) {//这一位添哪个数字 
                if (n > f[i - 1][g[p][j]]) {
                    n -= f[i - 1][g[p][j]];//如果添上该数字后仍然不够n就减去 
                } else {
                    break;
                }
            }
            printf("%d", j);//把添的这一位输出 
            p = g[p][j];//添了数字后到达下一个状态 
        }
        puts("");
    }
    return 0;
}