洛谷 P2657 [SCOI2009]windy数

题目：windy数

思路：数位dp

代码：

#include<bits/stdc++.h>
using namespace std;

#define read(x) scanf("%d",&x)

int L,R;
int a[20];
int f[20][20][2];

int dp(int x,int fa,int lmt) {
	if(x==0) return 1;
	if(fa!=-1&&f[x][fa][lmt]!=-1) return f[x][fa][lmt];
	int up=9;
	if(lmt) up=a[x];
	int ans=0;
	for(int i=0;i<=up;i++) {
		if(fa!=-1&&abs(i-fa)<2) continue;
		ans+=dp(x-1,(i==0&&fa==-1)?-1:i,lmt&&i==up);
	}
	if(fa!=-1) f[x][fa][lmt]=ans;
	return ans;
}

int slv(int x) {
	memset(f,-1,sizeof(f));
	int len=0;
	while(x) {
		a[++len]=x%10;
		x/=10;
	}
	return dp(len,-1,1);
}

int main() {
	read(L),read(R);
	printf("%d",slv(R)-slv(L-1));
	return 0;
}