BZOJ 1026: [SCOI2009]windy数【数位dp】

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

HINT

【数据规模和约定】

100%的数据，满足 1 <= A <= B <= 2000000000 。

思路:数位dp水题,dp[pos][last][lead]表示考虑到第pos位，上一位数字是last，上一位数字是不是前导0,然后随便乱搞一下

 1 #include<cstdio>
 2 int abs(int n){if(n<0)return -n;else return n;}
 3 using namespace std;
 4 int dp[11][11][2];
 5 int dfs(int pos,int last,int lead,int limit,int deci[])
 6 {
 7     if(pos==0)return 1;
 8     //if(limit==0 && dp[pos][last][lead]!=-1)return dp[pos][last][lead];
 9     int temp = limit ? deci[pos] : 9;deci[0] = 0 ;
10     for(int i = 0; i <= temp ; i++)
11     {
12         if(abs(i-last)>=2 || lead)
13         {
14             deci[0] += dfs(pos-1, i , i==0 && lead ,limit && i == temp ,deci);
15         }
16     }
17     if(limit==0)dp[pos][last][lead] = deci[0];
18     return deci[0];
19 }
20 int solve(int n)
21 {
22     int deci[11]={0};
23     while(n!=0)
24     {
25         deci[++deci[0]]=n%10;
26         n/=10;
27     }
28     return dfs(deci[0],0,1,1,deci);
29 }
30 int main()
31 {
32     int a,b;
33     for(int i=0;i<=10;i++)
34                 for(int j=0;j<=10;j++)
35                         for(int k=0;k<=1;k++)dp[i][j][k]=-1;
36     scanf("%d%d",&a,&b);
37     printf("%d",solve(b)-solve(a-1));
38     return 0;
39 }