HDU 5787 K-wolf Number（数位dp）

http://acm.split.hdu.edu.cn/showproblem.php?pid=5787

题意：
给出一个范围[l,r]和整数k，求出在该范围的数在十进制每k个相邻的数都不同的数的个数。

思路：

看题意就是挺明显的数位dp，一开始不想开5维数组，想用哈希记录状态，不过错了，可能是姿势不太对。。然后就开了5维数组，因为K最多就5嘛，注意好前导0的问题即可。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 1e6 + 5;

ll L,R;
int K;
int dig[30];
ll dp[30][11][11][11][11];

bool check(int pre1,int pre2,int pre3,int pre4,int x)
{
    if(K==2)  return x!=pre4;
    else if(K==3)  return x!=pre4 && x!=pre3;
    else if(K==4)  return x!=pre4 && x!=pre3 && x!=pre2;
    else return x!=pre4 && x!=pre3 && x!=pre2 && x!=pre1;
}

ll dfs(int pos, int pre1, int pre2, int pre3, int pre4, int lead ,int limit)
{
    if(pos==-1)  return lead!=1;
    if(!limit && dp[pos][pre1][pre2][pre3][pre4]!=-1)  return dp[pos][pre1][pre2][pre3][pre4];
    int up=limit?dig[pos]:9;
    ll tmp = 0;
    for(int i=0;i<=up;i++)
    {
        if(lead && i==0)  tmp+=dfs(pos-1,pre1,pre2,pre3,pre4,lead,limit&&i==dig[pos]);
        else if(check(pre1,pre2,pre3,pre4,i))
            tmp+=dfs(pos-1,pre2,pre3,pre4,i,0,limit&&i==dig[pos]);
    }
    if(!limit)  dp[pos][pre1][pre2][pre3][pre4]=tmp;
    return tmp;
}

ll solve(ll x)
{
    int pos=0;
    while(x)
    {
        dig[pos++]=x%10;
        x/=10;
    }
    return dfs(pos-1,10,10,10,10,1,1);
}

int main()
{
    //freopen("in.txt","r",stdin);
    while(~scanf("%lld%lld%d",&L,&R,&K))
    {
        memset(dp,-1,sizeof(dp));
        printf("%lld
",solve(R)-solve(L-1));
    }
    return 0;
}