HDU5787 K-wolf Number

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=5787

【题意】给定三个数i,j,k。问i-j之间有多少任意k位数都不相同的数。

【分析】数位dp，用dp[pos][p1][p2][p3][p4]表示当前位为第pos位，前第一位为p4，前第二位为p3，前第三位为p2，前第四位为p1的数的满足条件的数的个数。用特殊的10表示前导0。用dfs进行记忆化搜索。搜索过程中用flg标记前位是否取满以确定当前位的取值上限。

【代码】

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define LL long long
LL dp[20][12][12][12][12];
int dig[20];
LL l,r;
int k;
bool check(int p1,int p2,int p3,int p4,int p5){
if(k==2)
return p5!=p4;
if(k==3)
return (p5!=p4 && p5!=p3);
if(k==4)
return (p5!=p4 && p5!=p3 && p5!=p2);
return (p5!=p4 && p5!=p3 && p5!=p2 && p5!=p1);
}
LL dfs(int pos,int p1,int p2,int p3,int p4,bool flg){
if(pos==0)
return p4!=10;
if(!flg && dp[pos][p1][p2][p3][p4]!=-1)
return dp[pos][p1][p2][p3][p4];
int ed=flg?dig[pos]:9;
LL ans=0;
for(int i=0;i<=ed;++i){
if(i==0 && p4==10)
ans+=dfs(pos-1,10,10,10,10,flg && i==ed);
else
if(check(p1,p2,p3,p4,i))
ans+=dfs(pos-1,p2,p3,p4,i,flg && i==ed);
}
if(!flg)
dp[pos][p1][p2][p3][p4]=ans;
return ans;
}
LL cal(LL x){
if(x<=0)
return 0;
LL tmp=x;
int len=0;
while(tmp){
dig[++len]=tmp%10;
tmp/=10;
}
return dfs(len,10,10,10,10,1);
}
int main(){
while(~scanf("%I64d %I64d %d",&l,&r,&k)){
memset(dp,-1,sizeof(dp));
printf("%I64d\n",cal(r)-cal(l-1));
}
}