数位DP+二分搜索——HDU3943 K-th Nya Number

最新推荐文章于 2020-11-25 22:25:06 发布

三明自

最新推荐文章于 2020-11-25 22:25:06 发布

阅读量965

点赞数

分类专栏：数位DP

本文链接：https://blog.csdn.net/double215726/article/details/9383205

版权

数位DP 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=3943

题目大意：

给定一个左开右闭区间，(p，q]，求在该区间上第k个Nya Number，如果找不到则输出"Nya！"。

Nya Number的定义：含有x个4和y个7的数字。

比如：x=1，y=1，那么47，74，147。。。。都是Nya Number。

解题思路：

这个题目也是很经典的数位DP。

定义数组：dp[i][j][k]，表示一个共i位的数字中，含有j个4和k个7的数字的个数。

注意事项，就是p++，因为是左开区间，所以是不包括p的。

源代码：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<set>
#include<map>
#include<vector>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long LL;
LL dp[26][26][26];     //共i位有j个4,k个7
LL p,q,x,y;
int date[26];
void init()     //对dp数组进行预处理
{
    int i,j,k,t;
    memset(dp,0,sizeof(dp));
    dp[0][0][0]=1;
    for(i=1;i<=20;i++)
    {
        for(j=0;j<=i;j++)
        {
            for(k=0;k+j<=i;k++)
            {
                if(j-1>=0)      //第i位出现4
                    dp[i][j][k]+=dp[i-1][j-1][k];
                if(k-1>=0)      //第i位出现7
                    dp[i][j][k]+=dp[i-1][j][k-1];
                if(j+k<i)       //第i位出现其他数字
                    dp[i][j][k]+=dp[i-1][j][k]*8;
            }
        }
    }
    return;
}
LL DP(LL key)     //0--key-1出现了规定47数字的个数
{
    int i,j,k,t,len,four,seven;
    LL ans;
    memset(date,0,sizeof(date));
    len=0;
    while(key)
    {
        date[++len]=key%10;
        key/=10;
    }
    date[len+1]=0;
    ans=four=seven=0;
    for(i=len;i>=1;i--)
    {
        if(four>x || seven>y)    break;

        t=date[i];
        if(date[i]>4)   t--;
        if(date[i]>7)   t--;
        ans+=dp[i-1][x-four][y-seven]*t;                    //第i位不取4 7
        if(date[i]>4)   ans+=dp[i-1][x-four-1][y-seven];    //第i位取4
        if(date[i]>7)   ans+=dp[i-1][x-four][y-seven-1];    //第i位取7

        if(date[i]==4)  four++;
        if(date[i]==7)  seven++;
    }
    return ans;
}
LL solve(LL key)  //用二分搜索求取第key个数字
{
    LL l,r,a,b,mid;
    l=p;        //这个地方不需要+1，因为p作为全局变量，在主函数里面加过1了
    r=q;
    a=DP(p);
    while(r>l)
    {
        mid=(l+r)>>1;
        b=DP(mid+1);
        if(b-a<key)
            l=mid+1;
        else
            r=mid;
    }
    return r;
}
int main()
{
    freopen("in.txt","r",stdin);
    int cs,t,count,i,j;
    LL k,a,b;
    scanf("%d",&cs);
    count=0;
    init();
    while(cs--)
    {
        printf("Case #%d:\n",++count);
        scanf("%I64d %I64d %I64d %I64d",&p,&q,&x,&y);
        ++p;        //这个地方不要忘了，因为是左开区间，所以左边+1
        a=DP(p);
        b=DP(q+1);
        scanf("%d",&t);
        for(i=1;i<=t;i++)
        {
            scanf("%I64d",&k);
            if(a+k>b)
            {
                printf("Nya!\n");
                continue;
            }
            printf("%I64d\n",solve(k));
        }
    }
    return 0;
}