[HDOJ5787]K-wolf Number（数位DP）

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

题意：求[L,R]区间内的数字，使得所有长度为k的子数列内所有数位都不同。

K<=5的所以可以直接记录到前k个数字的值是多少。dp(l,p1,p2,p3,p4)分别记录就可以了。

弱智了WA了好几发，因为k每次取值是不一样的，所以dp数组还是应当放在while里清空。

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
const int maxn = 22;
int k;
int digit[maxn];
LL dp[maxn][12][12][12][12];
LL l, r;

bool ok(int a, int b, int c, int d, int e) {
    if(k == 2) return e != d;
    if(k == 3) return e != d && e != c;
    if(k == 4) return e != d && e != c && e != b;
    if(k == 5) return e != d && e != c && e != b && e != a;
}

LL dfs(int l, int p1, int p2, int p3, int p4, bool flag) {
    if(l == 0) return p4 != 10;
    if(!flag && ~dp[l][p1][p2][p3][p4]) return dp[l][p1][p2][p3][p4];
    LL ret = 0;
    int pos = flag ? digit[l] : 9;
    for(int i = 0; i <= pos; i++) {
        if(i == 0 && p4 == 10) ret += dfs(l-1,10,10,10,10,flag&&(i==pos));
        else if(ok(p1,p2,p3,p4,i)) ret += dfs(l-1,p2,p3,p4,i,flag&&(i==pos));
    }
    if(!flag) dp[l][p1][p2][p3][p4] = ret;
    return ret;
}

LL f(LL x) {
    if(l <= 0) return 0;
    int pos = 0;
    while(x) {
        digit[++pos] = x % 10;
        x /= 10;
    }
    return dfs(pos,10,10,10,10,true);
}

int main() {
    // freopen("in", "r", stdin);
    while(~scanf("%I64d%I64d%d",&l,&r,&k)) {
        memset(dp, -1, sizeof(dp));
        printf("%I64d
", f(r)-f(l-1));
    }
    return 0;
}