YbtOJ 数位DP G.幸运666

日常写点奇奇怪怪的乱搞做法 awa

这题跟前面几道数位 DP 的区别在于让求第 \(n\) 小的数。虽然我不会求也不想学这个，但我们可以 binary search！
问题就转换为求 \([1,mid]\) 中的幸运数个数，这样就和前面那些题做法一样了/xyx

code

#include<bits/stdc++.h>
#define int long long 
using namespace std;
const int N=40;
int a[N],vis[N][5];
int dfs(int pos,int op,bool limit,bool lead)
{
	if(pos<1) return (op==3);
	if((!limit)&&lead&&vis[pos][op]!=-1) return vis[pos][op];
	int up=9,ans=0;
	if(limit) up=a[pos];
	for(int i=0;i<=up;i++)
	{
		if(op==3) ans+=dfs(pos-1,3,limit&&(i==up),lead||i);
		else ans+=dfs(pos-1,(i==6?op+1:0),limit&&(i==up),lead||i);
	}
	if(!limit&&lead) vis[pos][op]=ans;
	return ans;
}
int work(int x)
{
	memset(vis,-1,sizeof(vis));
	int len=0;
	while(x)
	{
		a[++len]=x%10;
		x/=10;
	}
	return dfs(len,0,1,0);
}
int T;
signed main()
{
	scanf("%lld",&T);
	while(T--)
	{
		int x;
		scanf("%lld",&x);
		int l=666,r=1e10;
		while(l<r)
		{
			int mid=(l+r)>>1;
			if(work(mid)<x) l=mid+1;
			else r=mid;
		}
		cout<<l<<endl;
	}
	return 0;
}