UESTC 250 数位dp(数字相位数之间的差值不小于2)

http://acm.uestc.edu.cn/#/problem/show/250

windy定义了一种windy数。

不含前导零且相邻两个数字之差至少为 2 的正整数被称为windy数。

windy想知道,在 A B 之间,包括 A B ,总共有多少个windy数?

Input

包含两个整数, A   B

满足  1AB2000000000  .

Output

Sample input and output

Sample Input Sample Output
1 10
9

Source

Windy
/**
UESTC 250  数位dp(数字相位数之间的差值不小于2)
题目大意:求给定区间内的数字有多少满足一个数字的不同位数的数字之差不小于2
解题思路:数位dp,dp[i][j]表示i位数,以j开头满足条件的个数。将n分解入len数组,然后先统计位数小于len的个数(不能统计以0开头的),
           再然后统计位数正好为len的个数。最后不要忘了考虑n本身
*/
#include <string.h>
#include <algorithm>
#include <iostream>
#include <stdio.h>
using namespace std;
typedef long long LL;
LL dp[20][15];
int bit[20];
int abs(int x)
{
    if(x>0)
        return x;
    return -x;
}
void init()
{
    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<=j-2;k++)
            {
                dp[i][j]+=dp[i-1][k];
            }
            for(int k=j+2;k<=9;k++)
            {
                dp[i][j]+=dp[i-1][k];
            }
        }
    }
}

LL solve(LL n)
{
    if(n==0)return 0;
    int len=0;
    while(n)
    {
        bit[++len]=n%10;
        n/=10;
    }
    LL ans=0;
    for(int i=1;i<len;i++)
    {
        for(int j=1;j<=9;j++)
        {
            ans+=dp[i][j];
        }
    }
    for(int i=1;i<bit[len];i++)
    {
        ans+=dp[len][i];
    }
    int flag=1;
    for(int i=len-1;i>=1;i--)
    {
        for(int j=0;j<bit[i];j++)
        {
            if(abs(bit[i+1]-j)>=2)
                ans+=dp[i][j];
        }
        if(abs(bit[i+1]-bit[i])<2)
        {
            flag=0;
            break;
        }
    }
    if(flag)ans++;
    return ans;
}
int main()
{
    LL n,m;
    init();
    while(~scanf("%lld%lld",&n,&m))
    {
        printf("%lld\n",solve(m)-solve(n-1));
    }
    return 0;
}


你可能感兴趣的:(UESTC 250 数位dp(数字相位数之间的差值不小于2))