1026: [SCOI2009]windy数

数位dp,dp[i][j]中i,j分别表示位数和最高位数字

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<set>
#include<cmath>
#include<map>
#include<queue>
using namespace std;
int dp[15][15];//长度为i,最高位为j

void init(){
    memset(dp,0,sizeof(dp));
    for(int i=0;i<10;i++)
        dp[1][i]=1;
    for(int i=2;i<=11;i++)
        for(int j=0;j<10;j++)
            for(int k=0;k<10;k++){
                if(fabs(k-j)>=2)
                    dp[i][j]+=dp[i-1][k];
            } 
}


int solve(int n){
    int len=0;
    int a[15];
    while(n){
        a[++len]=n%10;
        n/=10;
    }
    a[len+1]=0;
    int ans=0;
    for(int i=1;i<len;i++)      //比他小至少一位的 
        for(int j=1;j<10;j++)
            ans+=dp[i][j];
    for(int i=1;i<a[len];i++)   //位数和其相同,但最小位小于他 
         ans+=dp[len][i];
    for(int i=len-1;i>=1;i--){  //最高一位和他相同 
        for(int j=0;j<a[i];j++)     //后面为0也可以 
            if(fabs(j-a[i+1])>1)
                ans+=dp[i][j];
        if(fabs(a[i+1]-a[i])<=1)
            break;
    }
    return ans;
}


int main(){
    int n,m;
    init();
    while(scanf("%d%d",&n,&m)==2){
        printf("%d\n",solve(m+1)-solve(n));
    }
    return 0;
}