HDU 5787 K-wolf Number

标签（空格分隔）： 数位DP

---

题目链接
记录一下相邻位置的数字,前导零特殊处理

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
typedef long long ll;
ll read()
{
	ll x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
int k,num[20];
ll l,r,f[20][11][11][11][11][2];
ll dp(int step,int a,int b,int c,int d,int lim)
{
	if(step==0)return 1;
	if(~f[step][a][b][c][d][lim])return f[step][a][b][c][d][lim];
	int x=lim?num[step]:9;
	ll res=0;
	for(int i=0;i<=x;i++)
	{
		if(k>=2&&i==d)continue;
		if(k>=3&&i==c)continue;
		if(k>=4&&i==b)continue;
		if(k>=5&&i==a)continue;
		res+=dp(step-1,b,c,d,d==10&&i==0?10:i,lim&&i==x);
	}
	return f[step][a][b][c][d][lim]=res;
}
ll calc(ll x)
{
	int len=0;
	while(x)num[++len]=x%10,x/=10;
	memset(f,-1,sizeof(f));
	return dp(len,10,10,10,10,1);
}
int main()
{
	while(~scanf("%lld%lld%d",&l,&r,&k))
	{
		printf("%lld\n",calc(r)-calc(l-1));
	}
	return 0;
}