HDU - 5787 K-wolf Number 数位DP

传送门

水人今天[u]多校打了一会就不行了，感觉自己都不会做。。。一直让队友单挑怪不好意思的，所以过了一会就跑掉了。。晚上回来补题发现有个数位DP，做了做，一开始因为细节问题没处理好wa了一发，改了之后过了。。。好感动。。原来我能做啊。。。= =[/u]

[u]lim：当前数字上限是否有限制，无限制为9,有限制为给出的数字 pos：当前位置 kk：标记当前已经选了几个数字，如果小于k个则为0,否则为1 mod：由于最大上限是5位，所以用一个模来存储 c:当前是第几位       pos是反的，c是正的。。恩
就这样 然后只要二维就够了，只存当前位置和那个模，然后注意细节，然后没了。
[/u]

[u][/u]#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
#include <vector>
#include <queue>
#include <algorithm>
#include <cstdlib>
using namespace std;
const int maxn = 31;
long long l,r;
int k,m,len;
int s[maxn];
long long d[maxn][100005];
long long cal(int lim,int pos,int kk,int mod,int c){
if(pos<0) return 1;
if(!lim&&d[pos][mod]!=-1) return d[pos][mod];
int up = 9;
if(lim) up = s[pos];
long long res = 0;
for(int i=0;i<=up;i++){
int tlim = 0,tmod = (mod*10);
if(lim&&i == up) tlim = 1;
if(pos == len-1) {
res += cal(tlim,pos-1,kk||(c+1)>=k,(tmod+i)%m,min(c+1,k));
continue;
}
int tt = tmod/10,flag = 1,tc = c;
while(tt){
tc --;
if(tc == 0) break;
if(i == tt%10) flag = 0;
tt /= 10;
}
if(flag) res += cal(tlim,pos-1,kk||(c+1)>=k,(tmod+i)%m,min(c+1,k));
}
if(!lim) d[pos][mod] = res;
return res;
}
int main(){
while(~scanf("%lld%lld%d",&l,&r,&k)){
memset(d,-1,sizeof d);
l --;
len = 0;
m = 1;
for(int i=0;i<k;i++)
m *= 10;
while(r){
s[len++] = r%10;
r /= 10;
}
long long t1 = cal(1,len-1,0,0,1);
len = 0;
while(l){
s[len++] = l%10;
l /= 10;
}
long long t2 = cal(1,len-1,0,0,1);
printf("%lld\n",t1-t2);
}
return 0;
}