hdu5787 数位dp

pos表示的是当前处理的是第几位,最低位是0位,abcd表示相邻的四位(一共五位) d是最后一位,就是最远的那位,主要熟悉递归流程,记忆化搜索
hdu5787 数位dp_第1张图片

#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
int K,num[22];
ll dp[22][11][11][11][11];
bool check(int p1,int p2,int p3,int p4,int u){
    if(K==2)    return u!=p4;
    else if(K==3)   return u!=p4 && u!=p3;
    else if(K==4)   return u!=p4 && u!=p3 && u!=p2;
    else return u!=p4 && u!=p3 && u!=p2 && u!=p1;
}
ll dfs(int pos,int p1,int p2,int p3,int p4,int flag){
//  cout <
//  cout << "flag = " << flag << endl;
//  system("pause");
    if(pos==0)   return p4!=10;
    if(!flag && ~dp[pos][p1][p2][p3][p4]) return dp[pos][p1][p2][p3][p4];
    ll res=0;
    int ed=flag?num[pos]:9;
    for(int u=0;u<=ed;u++){
        if(p4==10 && u==0)  res+=dfs(pos-1,10,10,10,10,flag&&ed==u);//只是最开始的时候会这样,到底返回后才能
//从1开始继续 , 然后1 10 10 10 0递归 结束了之后就变为 for循环+1  1 10 10 10 1了 
        else if(check(p1,p2,p3,p4,u))
            res+=dfs(pos-1,p2,p3,p4,u,flag&&ed==u);//因为递归是前面确定了后,处理后面的,是为了判断标志用的,所以
//应该是上一位到上限且这一位也到上限 
    }
    if(!flag)
        dp[pos][p1][p2][p3][p4]=res;
    return res;
}
ll solve(ll x){
    int len=0;
    while(x){
        num[++len]=x%10;
        x=x/10;
    }
    return dfs(len,10,10,10,10,1);
}
int main()
{
    ll l,r;
    while(~scanf("%lld%lld%d",&l,&r,&K)){
        memset(dp,-1,sizeof(dp));
    //  cout << solve(r) << endl;
        cout<1)<

你可能感兴趣的:(ACM,dp,数位dp)