洛谷 P2657[SCOI2009]windy 数

数位DP板子题。

DP[i][j]记录到第i位前一位数字是j的数字数。

下附代码：

#include<bits/stdc++.h>
using namespace std;
int dp[20][15];
int dim[20];
int dfs(int x,int st,int jb){
    if (!x) return 1;
    if (!jb && dp[x][st]!=-1) return dp[x][st];
    int maxn=9,ret=0;
    if (jb) maxn=dim[x];
    for (int i=0; i<=maxn; i++){
        if (abs(st-i)<2) continue;
        if (st==11 && i==0){
            ret+=dfs(x-1,11,(jb==1 && i==maxn));
        }
        else ret+=dfs(x-1,i,(jb==1 && i==maxn));
    }
    if (!jb) return dp[x][st]=ret;
    return ret;
}
int main(){
    string a,b;
    cin>>a>>b;
    memset(dp,-1,sizeof(dp));
    for (int i=0; i<a.length(); i++){
        dim[a.length()-i]=a[i]-'0';
    }
    dim[1]-=1;
    int j=1;
    while (dim[j]<0){
        dim[j]+=10;
        dim[j+1]-=1;
        j++;
    }
    int l=dfs(a.length(),11,1);
    for (int i=0; i<b.length(); i++){
        dim[b.length()-i]=b[i]-'0';
    }    
    int r=dfs(b.length(),11,1);
    printf("%d",r-l);
}