数位dp入门（第几个含有666的数）

求第n个含有666的数。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define B(x) (1<<(x))
typedef long long ll;
void cmax(int& a,int b){ if(b>a)a=b; }
void cmin(int& a,int b){ if(b<a)a=b; }
const int oo=0x3f3f3f3f;
const int MOD=1000000007;
const int maxn=5100;
const int maxm=21000;
ll dp[20][11][11][2];
int bit[20];
 
ll dfs(int pos,int p1,int p2,int is,int f){
    if(pos<1) return is;//数位dp模板
    if(!f&&dp[pos][p1][p2][is]!=-1) return dp[pos][p1][p2][is];
    int last= f ? bit[pos] : 9;
    ll res=0;
    for(int i=0;i<=last;i++){
        res+=dfs(pos-1,p2,i,is||(p1==6&&p2==6&&i==6),f&&i==last);
    }
    if(!f) dp[pos][p1][p2][is]=res;
    return res;//处理
}
 
ll Cnt(ll x){
    int len=0;
    while(x){
        bit[++len]=x%10;
        x/=10;
    }//先分解每个位置;
    return dfs(len,0,0,0,1);//处理1-x有多少个含有666的数
}
 
ll solve(int n){
    if(n>72915400)return (ll)10000000000;
    ll l=0,r=(ll)10000000000,mid,ans,t;
    while(l<r){
        mid=(l+r)>>1;
        t=Cnt(mid);//含有多少个
        if(t<n)
            l=mid+1;//如果1-mid间含有的666数t小于n,证明mid太小
        else
            r=mid-1,ans=mid;//ans记录mid数;
    }
    t=Cnt(ans);//再次看t的值
    if(t>=n){
        while(Cnt(ans)>=n)ans--;
        ans++;
    }else{
        while(Cnt(ans)<=n)ans++;
        ans--;
    }
    return ans;//最终就是所求数
}
 
int main(){
    //freopen("G:\\read.txt","r",stdin);
    //freopen("G:\\a.txt","w",stdout);
    int n,T;
    memset(dp,-1,sizeof dp);
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        printf("%I64d\n",solve(n));
    }
    return 0;
}