E - windy数

E - windy数

Time Limit:1000MS     Memory Limit:165888KB     64bit IO Format:%lld & %llu

Submit  Status

Description

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的入门题，题意是给出一个范围 A~B，求范围内windy数的个数，（windy数：一个数字中相邻两个数字相差必须大于等于2的数）。

如果用dp[i][j]来表示长度为i，最高位为j的windy数的个数，那么其实当我们要求位数更高（即i更大）的dp值的时候我们是可以用比他第一位的dp值转移过来的，因为其实对于这次转移我们仅仅需要考虑的是i位和i-1位的开头j相差是否为2.简单来说转移方程为

dp[i][j] += dp[i-1][k];   (if abs(k-j) >= 2)

然后就是在最后累计求和的时候，

例如 53579 这个数，应该先把 0~9999 的dp值求和

然后把 50000 ~ 49999 的dp值求和

最后再处理 50000~53579。

如果在处理的时候发现这个数已经不满足windy数了，那么就不需要再继续往下处理了。

#include"iostream"
#include"cstring"
#include"cstdio"

#define INF 0x7f7f7f7f
#define LL long long

using namespace std;

LL dp[15][10];            //dp[i][j]用来保存一共有i位数并且最高位为j的wendy数有多少个

int abs(int x)                //取绝对值
{
    if(x >= 0) return x;
    else return -x;
}

void init()                //初始化dp，
{
    memset(dp,0,sizeof(dp));
    for(int i = 0;i < 10;i++) dp[1][i] = 1;
    for(int i = 2;i < 12;i++)
    {
        for(int j = 0;j < 10;j++)
        {
            for(int k = 0;k < 10;k++)
            {
                if(abs(j - k) >= 2)
                    dp[i][k] += dp[i-1][j];
            }
        }
    }
}

int getnum(int x,int n)          //用来获取数x中第n位是什么
{
    for(int i = 1;i < n;i++)
        x /= 10;
    return x%10;
}

int getlen(int x)                //用来获取数x一共有多少位
{
    if(x == 0) return 1;
    int len = 0;
    while(x)
    {
        len++;
        x /= 10;
    }
    return len;
}

LL getans(int x)
{
    LL res = 0;
    int len = getlen(x);
    for(int i = 1;i < len;i++)             //首先将还不到最高位这个整数为止的ans求和
    {
        for(int j = 1;j < 10;j++)
        {
            res += dp[i][j];
        }
    }
    for(int i = 1;i < getnum(x,len);i++)      //将位数为最高位，但小于最高位的那个数的ans求和
    {
        res += dp[len][i];
    }
    for(int i = len-1;i > 0;i--)            //将最高位整数开始一直到x-1为止的ans求和
    {
        for(int j = 0;j < getnum(x,i);j++)
        {
            if(abs(getnum(x,i+1) - j) >= 2) res += dp[i][j];
        }
        if(abs(getnum(x,i+1) - getnum(x,i)) < 2) break;   //如果这个书已经被发现不满足wendy数，那么直接break
    }
    return res;
}

int main(void)
{
    int a,b;
    init();
    while(~scanf("%d%d",&a,&b))
    {
        LL ansa = getans(a);
        LL ansb = getans(b+1);             //因为实际getans(x)求的是 x-1 的ans，所以上一句求的是  ans(a-1)这一句求的是 ans(b)； 
        printf("%lld\n",ansb-ansa);
    }
    return 0;
}