P2657 [SCOI2009]windy数

思路清看代码

DFS版

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#define int long long
using namespace std ;
int read() {
	int x = 0 , f = 1 ; char s = getchar() ;
	while(s > '9' || s < '0') {if(s == '-') f = -1 ; s = getchar() ;}
	while(s <='9' && s >='0') {x = x * 10 + (s-'0'); s = getchar() ;}
	return x*f ;
}
int l , r ; 
int num[15] , f[15][15] ;
int dfs(int pos ,int pre ,int lead ,int limit) {
	if(!pos) return 1 ;
	if(!limit && !lead && f[pos][pre] != -1) return f[pos][pre] ; 
	int up = limit ? num[pos] : 9 ;
	int ans = 0 ;
	for(int i = 0 ; i <= up ; i ++) {
		if(abs(i-pre) < 2) continue ; 
		ans += dfs(pos-1,(!i&&lead) ? -2 : i,lead&&(i==0),limit&&(i == up)) ;
	}
	if(!limit&&!lead) f[pos][pre] = ans ;
	return ans ; 
}
int calc(int x) {
	int len = 0 ;
	while(x) {
		num[++len] = x % 10 ;
		x /= 10 ;	
	}
	memset(f,-1,sizeof (f)) ; 
	return dfs(len,-2,1,1) ;
}
signed main () {
	l = read() , r = read() ;
	cout << calc(r) - calc(l-1) << endl ;
	return 0 ;
}

循环版

#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 
*/