1026: [SCOI2009]windy数

1026: [SCOI2009]windy数

https://www.lydsy.com/JudgeOnline/problem.php?id=1026

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

 

Source

 

[Submit][Status][Discuss]

思路：数位dp经典题。详见代码。

state就是pre的意思。就是前一位。

对于前导0的问题，这里的处理方式就是如果有前导0的话就是往下dfs(因为之后可能就没有了)，或者没有前导0，并且abs（now-pre）>=2的话就向下dfs。这个地方初学者要仔细体会。

代码：

#include<bits/stdc++.h>
using namespace std;
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]!=-1)  
        return dp[pos][state];
    
    int ans=0;
    
    int up=limit?bit[pos]:9;
    for(int i=0;i<=up;i++)
    {
        if(abs(state-i)>=2 || lead)
        {
            ans+=dfs(pos-1,i,lead&&!i,limit&&i==bit[pos]);
            ///lead&&!i(有前导0，i是0.这个时候返回1，说明有前导0)
        }
    }
    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()
{
   int le,ri;
   memset(dp,-1,sizeof(dp));
   while(cin>>le>>ri)
   {
       printf("%d\n",solve(ri)-solve(le-1));
   }
   return 0;
}