启示录 poj3028

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/MrTinTin/article/details/82913091

数位DP。

#include<iostream>
#define f(i,l,r) for(i=(l);i<=(r);i++)
using namespace std;
long long f[22][4];
int T,n,m;
inline void MakeTable()
{
	int i,j,k;
	f[0][0]=1;
	f(i,1,21){
		f[i][0]=9*(f[i-1][0]+f[i-1][1]+f[i-1][2]);
		f[i][1]=f[i-1][0];
		f[i][2]=f[i-1][1];
		f[i][3]=f[i-1][2]+10*f[i-1][3];
	}
}
int main()
{
	int i,j,k;
	long long num;
	MakeTable();
	cin>>T;
	while(T--){
		cin>>n;
		for(m=3;f[m][3]<n;m++);
		k=0;
		for(i=m;i;i--){
			for(j=0;j<=9;j++){
				num=f[i-1][3];
				if(k==3){
					num+=f[i-1][0]+f[i-1][1]+f[i-1][2];
				}
				else if(j==6){
					if(k==1) num+=f[i-1][1]+f[i-1][2];
					else if(k==2) num+=f[i-1][0]+f[i-1][1]+f[i-1][2];
					else num+=f[i-1][2];
				}
				if(num<n){
					n-=num;
					continue;
				}
				if(k<3){
					if(j==6) k++;
					else k=0;
				}
				cout<<j;
				break;
			}
		}
		cout<<endl;
	}
	return 0;
}