数位DP bzoj1026

1026: [SCOI2009]windy数

Time Limit: 1 Sec  Memory Limit: 162 MB
Submit: 5809  Solved: 2589
[Submit][Status][Discuss]

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 。

代码：

 /*
 与上两道题类似，但不同的是本题要考虑首位非0。
 */
 #include<iostream>
 #include<string>
 #include<cstdio>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 #include<iomanip>
 #include<queue>
 #include<stack>
 using namespace std;
 int n,m;
 int dp[][];
 void init()
 {
     for(int i=;i<;i++)  //小于十的数单独计算
     {
         dp[][i]=;
     }
     for(int i=;i<;i++)
     {
         for(int j=;j<;j++)
         {
             for(int k=;k<;k++)
             {
                 if(abs(j-k)>=)
                 dp[i][j]+=dp[i-][k];
             }
         }
     }
 }
 int insum(int n)
 {
     int sum=;
     int cnt=;
     int c[]={};
     while(n)
     {
         c[++cnt]=n%;
         n=n/;
     }
     c[cnt+]=;
     for(int i=;i<cnt;i++)  //算前cnt-1位
     {
         for(int j=;j<;j++)   //j从1开始，前导非0
         {
             sum+=dp[i][j];
         }
     }
     for(int i=;i<c[cnt];i++)  //算最高位
     {
         sum+=dp[cnt][i];
     }
     for(int i=cnt-1;i>;i--)  //最高位位已经算过了，从cnt-1位开始算
     {
         for(int j=;j<c[i];j++)
         {
             if(abs(j-c[i+])>=)
             sum+=dp[i][j];
         }
         if(abs(c[i]-c[i+])<)
         break;
     }
     return sum;
 }
 int main()
 {
     while(cin>>n>>m)
     {
         memset(dp,,sizeof(dp));
         init();
         cout<<insum(m+)-insum(n)<<endl;
     }
     return ;
 }