HDU - 5787 K-wolf Number 数位DP

传送门

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

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

#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;
}