uestc1307 windy数 (数位DP）

题目

题意：　windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，
在A和B之间，包括A和B，总共有多少个windy数？

思路：
求长度小于len的windy数，长度等于len的windy数，这个数字本身的windy数。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;

int num[12];
int dp[12][10];

void init(){           //dp[i][j]代表长度为i,最高位为j的windy数个数 
    memset(dp,0,sizeof(dp));
    for(int i=0;i<10;i++) 
    dp[1][i]=1;
    for(int i=2;i<=10;i++){
        for(int j=0;j<10;j++){
            for(int k=0;k<10;k++){
                if(abs(j-k)>1)
                dp[i][j]+=dp[i-1][k];
            }
        }
    }
} 

int solve(int a){

    int len=0;
    while(a){          //将数字拆开 
        num[len++]=a%10;
        a/=10;
    }

    int ans=0;

    for(int i=1;i<len;i++){  //长度小于len的个数 
        for(int j=1;j<10;j++){
            ans+=dp[i][j];
        }
    }

    for(int j=1;j<num[len-1];j++){ //长度为len，但小于a的个数 
        ans+=dp[len][j];
    } 

    for(int i=len-1;i>=1;i--){    //处理a的个数 
        for(int j=0;j<num[i-1];j++){
            if(abs(j-num[i])>1) ans+=dp[i][j];
        }
        if(abs(num[i]-num[i-1])<=1) 
        break;
    }
    return ans;
}

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

}