P2657 [SCOI2009]windy数

思路清看代码

#include <bits/stdc++.h>
#define int long long
#define rep(x,y,z) for(int x = y ; x <= z ; x ++)
using namespace std ;
int f[20][20] , a[20] ;
int calc(int x) {
    int len = 0 ;
    while(x>0) {//江数位拆开以便于处理 
        a[++len] = x % 10 ;
        x /= 10 ;
    }int sum = 0 ;//用于统计答案 
    //先求len-1位的 
    for(int i = 1 ; i <= len-1 ; i ++) {//枚举位数 
        for(int j = 1 ; j <= 9 ; j ++) {
            sum += f[i][j] ;
        }
    }
    //然后求len位,最高位比len位的数小的 
    for(int i = 1 ; i < a[len] ; i ++) {
        sum += f[len][i] ;
    }
    //然后是len位,最高位与原数相同的 
    for(int i = len - 1 ; i >= 1 ; i --) {//枚举每一位 
        for(int j = 0 ; j <= a[i]-1 ; j ++) {//枚举i位上的数字 
            if(abs(j-a[i+1]) >= 2) {
                sum += f[i][j] ;//如果和上一位相同,就相加 
            }
        }if(abs(a[i+1]-a[i]) < 2) break ;
    }
    return sum ;
}
int n , m ;
signed main() {
    scanf("%lld%lld",&n,&m) ;
    rep(i,0,9) f[1][i] = 1 ;//初始化,[0,9]都是windy数                                 
    for(int i = 2 ; i <= 10 ; i ++) {//枚举位数 
        for(int j = 0 ; j <= 9 ; j ++) {//枚举最高位的数 
            for(int k = 0 ; k <= 9 ; k ++) {//枚举下一位的数 
                if(abs(k-j) >= 2) f[i][j] += f[i-1][k] ;
            }
        }
    }
    cout << calc(m+1) - calc(n) << endl ;
    return 0 ;
}
/*
f[i][j] , 位数为i最高位数字为j的个数
f[i][j] += f[i-1][k]  |k-j| >= 2 
*/