码迷,mamicode.com
首页 > 其他好文 > 详细

数位DP -启示录

时间:2019-04-27 13:16:20      阅读:138      评论:0      收藏:0      [点我收藏+]

标签:个数   数位dp   ==   oid   while   scan   code   数字   预处理   

 

http://poj.org/problem?id=3208

一个魔鬼数为包含连续三个666的的数字,给个n(n<5e7)求第n个魔鬼数。

预处理f[i][j],f[i][3]表示由前i位数字构成所有可能的魔鬼数,需要注意这里允许前导0存在。

f[i][2]表示由i位数字构成的开头为2个6的非魔鬼数个数,

f[i][1]表示由i位数字构成的开头为1个6的非魔鬼数个数,

f[i][1]表示由i位数字构成的开头为0个6的非魔鬼数个数,

之后从高位到低位填每个数,每个数从小到大枚举。

#include <cstdio>
#include <iostream>
using namespace std;
#define LL long long
LL f[18][5];
void init()
{
//    f[1][1]=1;
//    f[1][0]=9;
    f[0][0]=1;
    for(int i=1; i<=15; i++)
    {
        f[i][0]=(f[i-1][2]+f[i-1][1]+f[i-1][0])*9;
        f[i][1]=f[i-1][0];
        f[i][2]=f[i-1][1];
        f[i][3]=f[i-1][2]+f[i-1][3]*10;
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    init();
//    for(int i=1; i<=15; i++)
//        printf("%d ",f[i][3]);
    while(t--)
    {
        int n;
        scanf("%d",&n);
        int p;
        for(p=1; f[p][3]<n; p++);
        int cnk=0;
        for(; p>=1; p--)
        {
            for(int j=0; j<=9; j++)
            {
                int sum=0;
                if(j==6||cnk>=3)
                {
                    int t=cnk+1;
                    if(t>=3) sum+=f[p-1][0]+f[p-1][1]+f[p-1][2];
                    else if(t==2) sum+=f[p-1][1]+f[p-1][2];
                    else sum+=f[p-1][2];
                    sum+=f[p-1][3];
                }
                else
                    sum=f[p-1][3];
                if(sum<n)
                    n-=sum;
                else
                {
                    if(j==6)
                        cnk++;
                    else if(cnk<3)
                        cnk=0;
                    printf("%d",j);
                    break;
                }
            }
        }
        printf("\n");
    }
}

 

数位DP -启示录

标签:个数   数位dp   ==   oid   while   scan   code   数字   预处理   

原文地址:https://www.cnblogs.com/dongdong25800/p/10778150.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!