sdut 2609 A-Number and B-Number

http://acm.sdut.edu.cn/sdutoj/problem.php?action=showproblem&problemid=2609

数位DP 以前没怎么做过，自己憋了半天，还是看了标程，标程写的就是好呀。

相关注释见代码:

#include<iostream>
#include<cstdio>
#include<vector>
#include<stack>
#include<cstring>
#include<queue>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<set>
#include<map>

#define ll long long
#define ull unsigned long long
using namespace std;
const int INF=0x3f3f3f3f;
ll dp[22][10][2];
int a[22];//把这一个数按位分解
ll dfs(int n,int x,int flag,int limit)
//这里表示的意思应该是 到第n位的时候，前面构成的数的余数是x
//flag代表是否有 7 limit代表是否限制大小
{
    if(n==-1)//边界
    return (flag||x==0);
    if(!limit&&dp[n][x][flag]!=-1)
    return dp[n][x][flag];
    ll ans=0;
    int up=(limit)?a[n]:9;
    for(int i=0;i<=up;++i)//枚举当前位的取值
    {
        ans+=dfs(n-1,(x*10+i)%7,(flag||i==7),(limit&&i==a[n]));
    }
    if(!limit)//当没有限制的时候 就是记忆化搜索 有限制就是普通的dfs
    dp[n][x][flag]=ans;
    return ans;
}
ll solve(ull x)
{
    int len=0;
    while(x)
    {
        a[len++]=x%10;
        x=x/10;
    }
    return dfs(len-1,0,0,1)-1;
}
ll fNum(ull x)
{
    ull tmp=solve(x);
    return tmp-solve(tmp);
}
int main()
{
    //freopen("data.in","r",stdin);
    memset(dp,-1,sizeof(dp));
    ll k;
    while(cin>>k)
    {
        ull l=1,r=(ull)(pow(2.0,63)-1.0);
        while(l<=r)//由于没有思路，根本没想到二分
        {
            ull mid=(l+r)>>1;
            if(fNum(mid)>=k)
            r=mid-1;
            else
            l=mid+1;
        }
        cout<<l<<endl;
    }
    return 0;
}