洛谷P2657 [SCOI2009]windy数

洛谷P2657 [SCOI2009]windy数

本题是一道数位DP
看程序应该能看懂
中心思想：把一个数拆成每一位来处理
代码：

#include<bits/stdc++.h>
using namespace std;
int A,B;
int f[20][20];
void dp(){
	memset(f,0,sizeof(f));
	for (int i=0;i<10;i++) f[1][i]=1;
	for (int i=2;i<=10;i++){
		for (int j=0;j<10;j++){
			for (int k=0;k<10;k++){
				if (abs(j-k)>=2)
					f[i][j]+=f[i-1][k];
			}
		}
	}	
}
int work(int x){
	//把x分离出每一位： 
	int ans=0,len=0;
	int b[20];
	while (x>0){
		b[++len]=x%10;
		x/=10;
	}
	b[len+1]=-1;
	
	//位数是len-1位的windy数个数 ： 
	for (int i=1;i<len;i++)
		for (int j=1;j<10;j++)
			ans+=f[i][j];
	
	//位数是len位的但是首位小于x的首位的windy数个数： 
	for (int i=1;i<b[len];i++)	ans+=f[len][i];
	
	//位数是len位的并且首位等于x的首位的windy数个数：
	for (int i=len-1;i>0;i--){
		for (int j=0;j<b[i];j++)
			if (abs(j-b[i+1])>=2) ans+=f[i][j];
		if (abs(b[i]-b[i+1])<2) break;
	} 	 
	return ans;
}
int main(){
	scanf("%d%d",&A,&B);
	dp();
	printf("%d",work(B+1)-work(A));
}