首页 > 代码库 > poj 3208 Apocalypse Someday(数位dp)
poj 3208 Apocalypse Someday(数位dp)
题目链接: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;
}
poj 3208 Apocalypse Someday(数位dp)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。