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

最新推荐文章于 2019-03-12 16:30:04 发布

Dust_Heart

最新推荐文章于 2019-03-12 16:30:04 发布

阅读量343

点赞数

分类专栏：数位DP&记忆化搜索文章标签： BZOJ 1026 windy数 SCOI2009windy数

本文链接：https://blog.csdn.net/my_sunshine26/article/details/79725841

版权

数位DP&记忆化搜索专栏收录该内容

13 篇文章 0 订阅

订阅专栏

1026： [SCOI2009]windy数

Time Limit: 1 Sec Memory Limit: 162 MB

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 。

【题目链接】 link

【题意】

【思路】

用dp[i][j]表示长度为i，最高位为j的windy数个数，可以用暴力求得。

然后就是不断地分解输入的这个数，累加答案即可。

唯一值得注意的是，solve函数求出的是[1,x-1]内windy数的个数，因为每次都是从比当前位小的数去统计，没有考虑x是否为windy数。

具体细节见代码。

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
#define mst(a,b) memset((a),(b),sizeof(a))
#define rush() int T;scanf("%d",&T);while(T--)

typedef long long ll;
const int maxn = 15;
const ll mod = 1e9+7;
const ll INF = 1e18;
const double eps = 1e-6;

int digit[maxn];
ll dp[maxn][maxn];

void init()
{
    mst(dp,0);
    for(int i=0;i<=9;i++) dp[1][i]=1;
    for(int i=2;i<=10;i++)
    for(int j=0;j<=9;j++)
    for(int k=0;k<=9;k++)
    {
        if(abs(j-k)>=2) dp[i][j]+=dp[i-1][k];
    }
}

ll solve(ll x)
{
    ll ans=0;
    if(x==0) return 0;
    mst(digit,0);
    int len=0;
    while(x)
    {
        digit[++len]=x%10;
        x/=10;
    }
    for(int i=1;i<=digit[len]-1;i++) ans+=dp[len][i]; //长度相等，最高位小于
    for(int i=1;i<=len-1;i++)   //长度小于
    for(int j=1;j<=9;j++)
    {
        ans+=dp[i][j];
    }
    for(int i=len-1;i>0;i--)    //前面固定后，枚举每一位
    {
        for(int j=0;j<=digit[i]-1;j++)
        {
            if(abs(digit[i+1]-j)>=2) ans+=dp[i][j];
        }
        if(abs(digit[i+1]-digit[i])<2) break;
    }
    return ans;
}

int main()
{
    ll a,b;
    scanf("%lld%lld",&a,&b);
    init();
    printf("%lld\n",solve(b+1)-solve(a));
}