BZOJ 1026 (数位dp)

最新推荐文章于 2019-05-16 17:19:00 发布

nare123

最新推荐文章于 2019-05-16 17:19:00 发布

阅读量276

点赞数

CC 4.0 BY-SA版权

分类专栏： DP 文章标签： dp

本文链接：https://blog.csdn.net/nare123/article/details/78273481

DP 专栏收录该内容

17 篇文章

订阅专栏

本文详细解析了SCOI2009竞赛中的Windy数问题，介绍了一种通过记忆化搜索结合动态规划的方法来解决该问题，并提供了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1026: [SCOI2009]windy数

Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 8505 Solved: 3834
[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

【输出样例一】

【输出样例二】

20
HINT

【数据规模和约定】

100%的数据，满足 1 <= A <= B <= 2000000000 。
dp[pos][pre][stat][limit] 表示到第pos位的前一位是否有前导零，是否有限制
然后记忆化搜索

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <set>
#include <map>
#include <string>
#include <cstring>
#include <cmath>
using namespace std;
int a[100];
int dp[100][21][2][2];
int dfs(int pos,int pre,int stat,int limit){
    if (pos==-1)
    {
        return stat==0;
    }
    if (!limit&&dp[pos][pre+10][stat][limit]!=-1) return dp[pos][pre+10][stat][limit];
    int up=limit?a[pos]:9;
    int ans=0;
    if (!stat){
      for (int i=0;i<=up;i++)
          if (abs(i-pre)>=2) ans+=dfs(pos-1,i,0,limit&&i==a[pos]);
    }else 
    {
        for (int i=0;i<=up;i++){
            if (i==0) ans+=dfs(pos-1,-2,1,limit&&i==a[pos]);
            else ans+=dfs(pos-1,i,0,limit&&i==a[pos]);
        }
    }
    if (!limit) dp[pos][pre+10][stat][limit]=ans;
    return ans;
}
int solve (int x){
    int pos=0;
    while (x){
       a[pos++]=x%10;
       x/=10;
    }
    int ans=0;
    return dfs(pos-1,-2,1,1);
}
int main(){
    int a,b;
    memset(dp,-1,sizeof(dp));
    scanf("%d%d",&a,&b);
    printf("%d\n",solve(b)-solve(a-1));
}