poj 3208 Apocalypse Someday(数位dp)

来源：互联网发布：淘宝卖东西包装怎么办编辑：程序博客网时间：2024/05/22 21:41

题目链接：poj 3208 Apocalypse Someday

题目大意：给定n，输出第n大包含666的数字。

解题思路：数位dp，用类似AC自动机的思想进行转移。首先dp[i][j]表示说i位最后有j个连续6的情况数，这个预处理出

来。那么dp[i][3]即为i位有多少个满足的数。给定n，先确定位数d。然后从最高位向下判断，一开始肯定是需要3个连续

的6，所以u为3，然后根据后面添加的数字动态修改u值进行判断。

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;typedef long long ll;const int maxd = 10;const int status = 4;const int g[4][12] = {    {0, 0, 0, 0, 0, 0, 0, 0, 0, 0},    {3, 3, 3, 3, 3, 3, 0, 3, 3, 3},    {3, 3, 3, 3, 3, 3, 1, 3, 3, 3},    {3, 3, 3, 3, 3, 3, 2, 3, 3, 3}};ll n, dp[maxd+5][status];int main () {    memset(dp, 0, sizeof(dp));    dp[0][0] = 1;    for (int i = 1; i <= maxd; i++) {        dp[i][0] = (dp[i-1][0] + dp[i-1][1] + dp[i-1][2]) * 9;        dp[i][1] = dp[i-1][0];        dp[i][2] = dp[i-1][1];        dp[i][3] = dp[i-1][3] * 10 + dp[i-1][2];    }    int cas;    scanf("%d", &cas);    while (cas--) {        scanf("%lld", &n);        int d = 0, u = 3;        while (dp[d][3] < n)            d++;        while (d) {            ll k = 0;//            printf("%d:\n", d);            for (int i = 0; i < 10; i++) {                ll tmp = 0;                for (int j = 3; j >= g[u][i]; j--)                    tmp += dp[d-1][j];//                printf("n:%lld status: %d %d %lld\n", n, u, i, tmp);                if (k + tmp >= n) {                    printf("%d", i);                    u = g[u][i];                    break;                }                k += tmp;            }            n -= k;            d--;        }        printf("\n");    }    return 0;}

0 0