C - Apocalypse Someday (数位dp+二分）

C - Apocalypse Someday

 

题意：找第n个含有666的数。很6啊。。。。

思路;其他都好说，就是想二分问题上出了点小问题。。。这个题和基本的题型找出含有47的数字的数量差不多，这个是找出含有666的数字的数量。但是这个题不是问的数量有多少，而是问的第n大个，那个数量已经固定了，只需要算出1到某个数含有666的数量最接近n的那个数是什么，即可。也就是用一下二分，二分所有数，找出一个数量是小于n的最大的那个就行了。这个是问题关键部分也是重要部分

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long LL;
#define INF 0xfffffffffLL
int a[22];
LL dp[22][2][2][2];
LL dfs(int pos,int sta1,int sta2,int k,int limit)
{
    if(pos==-1)
	return k;
    if(!limit&&dp[pos][sta1][sta2][k]!=-1)
	return dp[pos][sta1][sta2][k];
    int up=limit?a[pos]:9;
    LL ans=0;
    for(int i=0;i<=up;i++)
    {
        if(k==1)
		ans+=dfs(pos-1,i==6,sta1,1,limit&&i==up);
        else 
		ans+=dfs(pos-1,i==6,sta1,sta1&&sta2&&i==6,limit&&i==up);
    }
    if(!limit)dp[pos][sta1][sta2][k]=ans;
    return ans;
}
LL cal(LL x)
{
    int pos=0;
    while(x)
    {
        a[pos++]=x%10;
        x/=10;
    }
    return dfs(pos-1,0,0,0,1);
}
int main()
{   int t;
    LL n;
    memset(dp,-1,sizeof(dp));
    scanf("%d",&t);
    while(t--)
    {
        scanf("%lld",&n);
        LL l=1,r=INF,mid;
        while(l<=r)
        {
            mid=(l+r)/2;
            if(cal(mid)<n)
            {
                l=mid+1;
            }
            else
            {
                r=mid-1;
            }

        }
       printf("%lld\n",l);
    }


    return 0;
}