Description
windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道,
在A和B之间,包括A和B,总共有多少个windy数?
Solution
dp[i][j] i位数中有多少个以j开头的windy数
work(n)是求1~n-1有多少个windy数 n是一个len位数,用bit[i]储存第i位上的数字
可以分为几部分:
1.1~len-1位数中有多少个windy数
2.len位数中最高位与n不同的数
3.len位数中最高位与n相同的数(需要一位一位地推)
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> typedef long long LL; using namespace std; LL a,b,dp[15][10],bit[15]; void init() { for(int i=0;i<=9;i++) dp[1][i]=1; 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) dp[i][j]+=dp[i-1][k]; } LL work(LL n) { LL res=0;int len=0; while(n) { bit[++len]=n%10; n/=10; } for(int i=len-1;i>=1;i--) { for(int j=1;j<=9;j++) res+=dp[i][j]; } //part 1 for(int i=1;i<=bit[len]-1;i++) res+=dp[len][i]; //part 2 for(int i=len-1;i>0;i--) { for(int j=0;j<=bit[i]-1;j++) { if(abs(bit[i+1]-j)>=2) res+=dp[i][j]; } if(abs(bit[i]-bit[i+1])<2)break; } //part 3 return res; } int main() { init(); scanf("%lld%lld",&a,&b); printf("%lld ",work(b+1)-work(a)); return 0; }