HDU 5787 K-wolf Number

题目分析

多校的题目，状态比较好想，但是真的不好写，需要用一些比较巧妙的方法。这道题是数位dp，对于题目要求，我们需要定义的状态时，第一维表示当前处理的数的位数，后面的4维表示当前数字的前4位数，接下来表示当目前为止所有数字位上的数字是否满足K-wolf Number，最后一维肯定还是是否有限制了。对于这道题处理上的小技巧就是：1：当你处理前几位的时候，他们前面是没有数字的，因此我们将他们前面的数字统一初始化为10，这样一来后面的就绝对不会跟前面的相等，因为后面的只可能在0~9之间。2：就是对于前导0的判断，很明显，如果数字前面均是0的话，很明显前导0是不算的，这样一来就不好处理，我们可以这样判断，如果该为之前的数字是10（说明前面没有数字），同时这一位为0，那么很明显我们就可以就可以将这一位表示为10，这样就可以处理了，代码写起来也很简单，就是普通的数位dp模板的写法。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
const int maxn = 20;

LL dp[maxn][15][15][15][15][2],bit[maxn],L,R,K;

LL dfs(int pos,int pre1,int pre2,int pre3,int pre4,int same,int limit){
    if(pos < 1) return same == 0;
    LL &ans = dp[pos][pre1][pre2][pre3][pre4][same];
    if(!limit && ans != -1) return ans;
    LL ret = 0;
    int len = limit?bit[pos]:9;
    for(int i = 0; i <= len; i++){
        int temp = 1;
        if(K == 5){
            if(i != pre1 && i != pre2 && i != pre3 && i != pre4)
                temp = 0;
        }
        else if(K == 4){
            if(i != pre2 && i != pre3 && i != pre4)
                temp = 0;
        }
        else if(K == 3){
            if(i != pre3 && i != pre4)
                temp = 0;
        }
        else{
            if(i != pre4)
                temp = 0;
        }
        ret += dfs(pos-1, pre2, pre3, pre4, (pre4==10&&i==0)?10:i, same||temp, limit&&i==len);
    }
    if(!limit) ans = ret;
    return ret;
}

LL solve(LL n){
    int len = 0;
    while(n){
        bit[++len] = n%10;
        n /= 10;
    }
    return dfs(len, 10, 10, 10, 10, 0, 1); //前面4位数字分别是什么，10表示没有数字,是否有存在相邻的位数字中有相同的，是否有限制
}

int main(){
    while(scanf("%I64d %I64d %I64d", &L, &R, &K) != EOF){
        memset(dp, -1, sizeof(dp));
        printf("%I64d\n", solve(R) - solve(L-1));
    }
    return 0;
}