bzoj1026: [SCOI2009]windy数 [简单数位dp]

考完联考就一直颓啊颓，根本写不动题目啊。

总算过了一道题，顿时有种便秘以后通畅的感觉。。

这道题算是比较简单的数位dp了，但是状态不好弄了一个晚上，说起来都是泪啊。

首先a到b之间满足条件的数，就是sum(b)-sum(a-1)有木有

然后为了服务群众，还是稍微详细点讲一下dp的过程吧。

状态: f[i][j]表示i位数字最高位为j的满足题设条件的方案数。

预处理f[i][j]还是挺水的~f[i][j]=Sigma(f[i-1][k])  abs(k-j)>=2

然后重要的东西来了，对于sum(b)

假设b的位数为n，表示为 anan-1...a1

那么我们先将f[i][j]  (1<=i<=n-1, 1<=j<=9)累加，加到ans里先，因为位数比b小，毫无疑问在[0,b]之间

然后对于[10^(n-1) ， anan-1...a1]，直接套用f[i][1...9]显然是不科学的。

所以我们这样玩：(以下1....x部分是需要用和前一项绝对值超过2判断的)

[10^n-1,an...00)       f[n][1...(an   -1)]

[an....00,anan-1...00)     f[n-1][1...(an-1   -1)]

同理。。

需注意到一个地方，如果出现[an....00,anan-1...00) 却abs(an-an-1)<2的情况，那么我们需要直接break

因为接下来方案的计算是建立在固定下来的前缀(此处为an) 合法的前提下。

不合法却继续做之后的统计并加入答案，显然不对。

另外，由于我们最后加出来的结果为[0,b)内的方案数，因此sum(b)应改为sum(b+1)

------------------------------------------------------------------------------------------------------------------------------------------------

Code:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;

ll f[13][13],t[13],a,b;

ll dp(ll x){
	ll ret=0,cnt=0;
	while(x){
		t[++cnt]=x%10;
		x/=10;
	}
	for(ll i=1;i<cnt;i++)
	for(ll j=1;j<=9;j++)
		ret+=f[i][j];
	for(ll i=1;i<t[cnt];i++) ret+=f[cnt][i];
	for(ll i=cnt-1;i;i--){
		for(ll j=0;j<t[i];j++)
			if(abs(j-t[i+1])>=2) ret+=f[i][j];
//		if(abs(t[i+1]-t[i])<2) break;
	}
	return ret;
}

int main(){
	freopen("1026.in","r",stdin);
	scanf("%lld%lld",&a,&b);
	for(ll i=0;i<=9;i++) f[1][i]=1;
	for(ll i=2;i<=12;i++)
	for(ll j=0;j<=9;j++)
	for(ll k=0;k<=9;k++)
		if(abs(k-j)>=2) f[i][j]+=f[i-1][k];
	printf("%lld\n",dp(b+1)-dp(a));
	return 0;
}