UESTC OJ Windy 数

Windy 数

【题目描述】

windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道，在A和B之间，包括A和B，总共有多少个windy数？

【输入】

包含两个整数，A B。

【输出】

一个整数。

【输入样例一】

1 10

【输出样例一】

9

【数据规模和约定】

100%的数据，满足 1 <= A <= B <= 2000000000 。

数位DP；dp[i][j]:位数为i最高位为j的windy数的个数。

 

#include<iostream>

 using namespace std;
 #include<string.h>
 #include<math.h>
int dp[11][10];
int fun(int n)
 {
   int dight[11],len=1,i,j;
   int sum=0;
   while(n)
   {
     dight[len++]=n%10;
     n=n/10;
   }
   for(i=1;i<len-1;i++)
   for(j=1;j<10;j++)
   sum=sum+dp[i][j];//例如n=363645，计算1<=n<=99999的Windy 数个数
   for(j=1;j<dight[len-1];j++)
    sum=sum+dp[len-1][j];//例如n=363645，计算100000<=n<=299999的Windy 数个数
   for(i=len-2;i>0;i--)
   {
       for(j=0;j<dight[i];j++)
        if(fabs(j-dight[i+1])>=2)
        { 
            sum=sum+dp[i][j];//例如n=363645，计算300000<=n<363645的Windy 数个数
        }
       if(fabs(dight[i]-dight[i+1])<2)
          break;
   }
   return sum;
 }
 int main()
 {
     int i,j,k,t;
     int ans;
     int n,m;
      memset(dp,0,sizeof(dp));
      for(i=0;i<10;i++)
         dp[1][i]=1;//为什么0~9是Windy 数？？？我有疑问。
         for(i=1;i<11;i++)
            for(j=0;j<=9;j++)
            {
                for(k=0;k<=9;k++)
                if(fabs(j-k)>=2)
               dp[i][j]=dp[i][j]+dp[i-1][k];
            }
          cin>>n>>m;
         ans=fun(m+1)-fun(n);
         cout<<ans<<endl;
     return 0;
 }