Windy数(数位dp)

来源:互联网 发布:软件著作权使用说明书 编辑:程序博客网 时间:2024/05/19 15:21

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

Input

  包含两个整数,A B。

Output

  一个整数

Sample Input
【输入样例一】1 10【输入样例二】25 50
Sample Output
【输出样例一】9【输出样例二】20
Hint

【数据规模和约定】

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



数据很大,用数位dp。

源代码:

#include <iostream>#include <cstring>#include <algorithm>#include <cmath>#include <stdio.h>using namespace std;int dp[20][10];int a[20];int digit(long long x){    memset(a,0,sizeof(a));    int length = 0;    while (x){        a[++length] = x % 10;      //  cout << a[length];        x /= 10;    }    //cout << endl;    return length;}long long solve (long long x){    long long ans = 0;    int length = digit(x);    //for (int i = 1; i <= length; i++)      //  cout << a[i];    //cout << endl;    for (int i = 1; i <= a[length]-1; i++)        ans += dp[length][i];    for (int i = length - 1; i > 0; i--)        for (int j = 1; j <= 9; j++)            ans += dp[i][j];    for (int i = length - 1; i > 0; i--){        for (int j = 0; j <= a[i]-1; j++)            if (abs(a[i+1]-j)>=2)                ans += dp[i][j];        if (abs(a[i+1] - a[i]) < 2)            break;    }    return ans;}int main(){    memset(dp,0,sizeof(dp));    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(j - k) >= 2)                    dp[i][j] += dp[i-1][k];    long long a,b;    scanf("%lld%lld",&a,&b);    printf("%lld\n",solve(b+1)-solve(a));    return 0;}



原创粉丝点击