BZOJ 1026, Windy数

Problem

Mean

Analysis

Code

#include<cstdio>
int a,b,dp[11][11];
int abs(int a){return a<0?-a:a;}
int solve(int x){
int len=0,ans=0,dig[11];
while(x){
dig[++len]=x%10;
x/=10;
}
for(int i=1;i<len;i++) for(int j=1;j<10;j++) ans+=dp[i][j];
for(int i=len;i;i--){
if(i==len) for(int j=1;j<dig[i];j++) ans+=dp[i][j];
else{
for(int j=0;j<dig[i];j++) if(abs(j-dig[i+1])>1) ans+=dp[i][j];
if(abs(dig[i]-dig[i+1])<2) break;
}
}
return ans;
}
int main(){
for(int i=0;i<10;i++) dp[1][i]=1;
for(int i=2;i<11;i++)
for(int j=0;j<10;j++)
for(int k=0;k<10;k++) if(abs(k-j)>1) dp[i][j]+=dp[i-1][k];
scanf("%d%d",&a,&b);
printf("%d",solve(b+1)-solve(a));
return 0;
}