BZOJ 1026: [SCOI2009]windy数

题目在这里呀!
看到这道题题目的时候以为这是一道很简单的题很简单的很简单很简很…
嗯结果做了好长时间啊qwq 可能是我对记忆化搜的数位dp情有独钟??


题意

emmm题目很短了不需要概括了吧qaq?

题解

一开始写了一个非常简单的数位dp,发现一开始的0没有处理好,然后就脑子一抽改啊改越改越错,最后终于从头开始,那么可以说就是枚举这个数字有几位,然后做数位dp即可,剩下就很简单了,我为什么不去想想枚举啊啊啊。

//Suplex
#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
int A,B,len,digit[1000],dp[1000][15];

int dfs(int pos,int pre,int flag)
{
    if(!pos) return 1;
    if(!flag && dp[pos][pre] != -1) return dp[pos][pre];
    int sum=0,top=9;
    if(flag) top=digit[pos];
    for(int i=0;i<=top;i++) if(abs(i-pre)>=2) sum+=dfs(pos-1,i,flag && i==top);
    if(!flag) dp[pos][pre]=sum;
    return sum;
}

int solve(int x)
{
    if(x==0) return 0;
    len=0;
    while(x){
        digit[++len]=x % 10;
        x/=10;
    }
    int ans=0;
    for(int i=len;i;i--){
        int top=9;
        if(i==len) top=digit[len];
        for(int j=1;j<=top;j++) ans+=dfs(i-1,j,(i==len) && (j==top));
    }
    return ans;
}

int main()
{
    scanf("%d%d",&A,&B);
    memset(dp,-1,sizeof(dp));
    printf("%d\n",solve(B)-solve(A-1));
    return 0;
}

猜你喜欢

转载自blog.csdn.net/leo_nasir/article/details/81144530