题目分析
多校的题目,状态比较好想,但是真的不好写,需要用一些比较巧妙的方法。这道题是数位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;
}