bzoj 1026: [SCOI2009]windy数 数位dp

1026: [SCOI2009]windy数

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 6376   Solved: 2849
[ 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 。

解法：dp[pos][state]表示[pos...0]位，并且第pos+1位数位state的种数。

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;

#define all(x) (x).begin(), (x).end()
#define for0(a, n) for (int (a) = 0; (a) < (n); (a)++)
#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
#define mes(a,x,s)  memset(a,x,(s)*sizeof a[0])
#define mem(a,x)  memset(a,x,sizeof a)
#define ysk(x)  (1<<(x))
typedef long long ll;
typedef pair<int, int> pii;
const int INF =0x3f3f3f3f;

int bit[11],dp[11][10];


int dfs(int pos,int state,int lead,int limit)
{
    if(pos==-1) return 1;
    if(!lead&&!limit&&~dp[pos][state])  return dp[pos][state];
    int ans=0;

    int up=limit?bit[pos]:9;
    for0(i,up+1) if(lead||abs(state-i)>=2)
    {
        ans+=dfs(pos-1,i,lead&&!i,limit&&i==bit[pos]);
    }
    if(!lead&&!limit)  dp[pos][state]=ans;
    return ans;

}
int solve(int x)
{
    int nbit=0;
    while(x)
    {
        bit[nbit++]=x%10;
        x/=10;
    }
    return dfs(nbit-1,-1,1,1 );//把0算作windy数
}

int main()
{
   std::ios::sync_with_stdio(false);
   int le,ri;
   mem(dp,-1);
   while(cin>>le>>ri)
   {
       printf("%d\n",solve(ri)-solve(le-1));
   }
   return 0;
}