BZOJ1026

1026: [SCOI2009]windy数

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 2254   Solved: 981
[ Submit][ Status]

Description

windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，在A和B之间，包括A和B，总共有多少个windy数？

Input

包含两个整数，A B。

Output

一个整数。

Sample Input

【输入样例一】
1 10

【输入样例二】
25 50

Sample Output

【输出样例一】
9

【输出样例二】
20

【数据规模和约定】
20%的数据，满足 1 <= A <= B <= 1000000 。
100%的数据，满足 1 <= A <= B <= 2000000000 。

【题解】数位DP,具体见代码实现。

#include<cstdio>
#include<cstring>

inline int abs(int x){return x<0?(-x):x;}

int a[20],b[20];
long long f[20][10]; 

int main()
{
	
	int A,B;scanf("%d%d",&A,&B);
	int la=0,lb=0;
	while(A){
		a[++la]=A%10;A/=10;
	}
	B++;
	while(B){
		b[++lb]=B%10;B/=10;
	}

	memset(f,0,sizeof(f));
	for(int i=0;i<=9;i++)f[1][i]=1;
	for(int i=2;i<=lb;i++)
	 	for(int j=0;j<=9;j++)
	 		 for(int k=0;k<=9;k++)
			   if(abs(j-k)>1)f[i][j]+=f[i-1][k];

	long long vb=0,va=0;
	//vb情况； 
	for(int i=0;i<b[lb];i++)vb+=f[lb][i];
	for(int i=lb-1;i;i--)vb+=f[i][1]+f[i][0];
	for(int i=lb-1;i;i--)
	{
		 for(int j=0;j<b[i];j++)
			  if(abs(b[i+1]-j)>1)vb+=f[i][j];
		 if(abs(b[i+1]-b[i])<2)break;	
	}
	//va情况;
	for(int i=0;i<a[la];i++)va+=f[la][i];
	for(int i=la-1;i;i--)va+=f[i][1]+f[i][0];
	for(int i=la-1;i;i--)
	{
		 for(int j=0;j<a[i];j++)
	 		 if(abs(a[i+1]-j)>1)va+=f[i][j];
	 	 if(abs(a[i+1]-a[i])<2)break;
	} 
	printf("%lld\n",vb-va);
	return 0;	
}