BZOJ1026: [SCOI2009]windy数(数位dp的三种解决方式)


BZOJ1026: [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

   1 10
 

Sample Output 1

  9
    

Sample Input 2

   25 50
 

Sample Output 2

  20
  

HINT

题目地址:BZOJ1026: [SCOI2009]windy数

题目大意: 题目很简洁了:)

  方法一:dp 多一维状态表示是否紧贴上界,看情况再多一维表示是否有前导0
  优点:容易想(容易口糊)
  缺点:写起来不一定简单
     多组询问原来 dp 的结果废了,每次都要重新清空再做一遍
     有些数据组数多的题过不去


AC代码

#include <cstdio> 
#include <cstring>
#include <algorithm>
using namespace std;
int A,B;
int a[11],dp[11][10][2][2];
int solve(int x){
    if(!x)return 0;
    int len=0;
    while(x){
        a[++len]=x%10;
        x/=10;
    }
    for(int i=1;i<=len/2;i++)
        swap(a[i],a[len-i+1]);
    memset(dp,0,sizeof(dp));
    //dp[i][j][k][l] 第i位,取了j,k=1表示有前导0,l=1表示贴上界 
    dp[1][0][1][0]=1;
    for(int i=1;i<a[1];i++)
        dp[1][i][0][0]=1;
    dp[1][a[1]][0][1]=1;
    for(int i=1;i<len;i++){
        //有前导0
        if(dp[i][0][1][0]){
            dp[i+1][0][1][0]+=dp[i][0][1][0];
            for(int k=1;k<=9;k++)
                dp[i+1][k][0][0]+=dp[i][0][1][0];
        }
        //没前导0
        for(int j=0;j<=9;j++){
            if(dp[i][j][0][0])
                for(int k=0;k<=9;k++)
                    if(abs(j-k)>=2)
                        dp[i+1][k][0][0]+=dp[i][j][0][0];
            if(dp[i][j][0][1])
                for(int k=0;k<a[i+1];k++)
                    if(abs(j-k)>=2)
                        dp[i+1][k][0][0]+=dp[i][j][0][1];
        }
        //贴上界
        if(abs(a[i]-a[i+1])>=2) 
            dp[i+1][a[i+1]][0][1]+=dp[i][a[i]][0][1];
    }
    int res=0;
    for(int j=0;j<=9;j++)
        res+=dp[len][j][0][0];
    res+=dp[len][a[len]][0][1];
    return res;
}
int main(){
    scanf("%d%d",&A,&B);
    printf("%d\n",solve(B)-solve(A-1));
    return 0;
}

  方法二:dp 预处理不贴边界的情况,然后暴力做出贴边界的情况
  优点:时间复杂度和空间复杂度都很优秀
  缺点:思维难度高


AC代码

#include <cstdio>
#include <algorithm>
using namespace std;
int A,B;
int dec[11],dp[11][10];
int f(int x){
    if(!x)return 0;
    int res=0,len;
    for(int i=10;i>=1;i--)
        if(x>=dec[i]){
            len=i;
            break;
        }
    for(int i=1;i<len;i++)
        for(int j=1;j<=9;j++)
            res+=dp[i][j];
    int now,pre;
    now=x/dec[len];
    for(int i=1;i<now;i++)
        res+=dp[len][i];
    x=x%dec[len];len--;
    pre=now;
    while(len){
        now=x/dec[len];
        for(int i=0;i<now;i++)
            if(abs(i-pre)>=2)res+=dp[len][i];
        if(len==1)
            if(abs(now-pre)>=2)res+=dp[1][now];
        if(abs(now-pre)<2)break;
        x=x%dec[len];len--;
        pre=now;
    }
    return res;
}
int main(){
    scanf("%d%d",&A,&B);
    dec[1]=1;
    for(int i=2;i<=10;i++)
        dec[i]=dec[i-1]*10;
    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];
    printf("%d\n",f(B)-f(A-1));
    return 0;
}

  方法三:带备忘录的dp(记忆化)
  优点:时间复杂度和空间复杂度都很优秀,代码复杂度低
  缺点:时间复杂度略高于方法二(会被卡)


AC代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int A,B;
int a[11],dp[11][10];
int calc(int k,int pre,int cas){
    if(!k)return 1;
    if(!cas && dp[k][pre]!=-1)return dp[k][pre];
    int res=0,top=cas?a[k]:9;
    for(int i=0;i<=top;i++)
        if(abs(i-pre)>=2)
            res+=calc(k-1,i,cas && i==top);
    if(!cas)dp[k][pre]=res;
    return res;
}
int solve(int x){
    if(!x)return 0;
    int len=0;
    while(x){
        a[++len]=x%10;
        x/=10;
    }
    int res=0;
    for(int i=len;i>=1;i--){
        int top=(i==len)?a[len]:9;
        for(int j=1;j<=top;j++)
            res+=calc(i-1,j,(i==len) && (j==top));
    }
    return res;
}
int main(){
    scanf("%d%d",&A,&B);
    memset(dp,-1,sizeof(dp));
    printf("%d\n",solve(B)-solve(A-1));
    return 0;
}


  作者:skl_win
  出处:https://www.cnblogs.com/shaokele/
  本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。

posted @ 2018-10-24 23:28  skl_win  阅读(...)  评论(... 编辑 收藏
Live2D