题目链接:
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;
}