[BZOJ1026]SCOI2009 windy数|数位DP

第一次见数位DP，看完题解才会做，f[i][j]表示最高有i位，第i位是j的个数，然后枚举0-9转移状态就行了。。计数的时候要考虑全面。。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<memory.h>
#define ll long long
using namespace std;
int i,j,k;
ll a,b,f[20][20];
ll cnt(ll x)
{
	int i,j,k,len=0,w[20];
	ll ans=0ll;
	while (x) w[++len]=x%10,x/=10;
	if (!len) return 0;
	for (i=1;i<len;i++)
		for (j=1;j<10;j++)
			ans+=f[i][j];
	for (i=1;i<w[len];i++) ans+=f[len][i];
	for (i=len-1;i;i--)
	{
		for (j=0;j<w[i];j++) 
			if (abs(w[i+1]-j)>1) ans+=f[i][j];
		if (abs(w[i+1]-w[i])<2) break;
	}
	if (!i&&abs(w[1]-w[2])>1) ans++;
	return ans;
}
int main()
{
	cin>>a>>b;
	memset(f,0,sizeof(f));
	for (i=0;i<10;i++) f[1][i]=1;
	for (i=2;i<=11;i++)
		for (j=0;j<10;j++)
			for (k=0;k<10;k++)
				if (abs(j-k)>1) f[i][j]+=f[i-1][k];
	cout<<cnt(b)-cnt(a-1);
}