Bzoj 1026 windy数【数位dp】

1026: [SCOI2009]windy数

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 5697   Solved: 2544
[ Submit][ Status][ Discuss]

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 。

思路：

1、经典数位dp T T QAQ调了好久

2、设dp【i】【j】表示长度为i的数字中，最高位数字为j有多少符合的情况。

例如dp【1】【1】=7；分别表示：13 14 15 16 17 18 19

那么状态转移方程：

dp【i】【j】=sigma（dp【i-1】【k】{abs（j-k）>=2}）

3、处理完dp【】【】数组之后呢，我们要求a到b区间内的个数，我们不妨计算出0到a区间内个数和0到b区间内个数，然后做差即可。

4、说起来很容易，不经常敲的类型题其实突然想一个人完美实现还是需要点时间的 T T大家尽量自己尝试调试。具体实现费劲的可以参考Ac代码：

#include<stdio.h>
#include<string.h>
using namespace std;
int dp[12][10];//dp[i][j]表示的是长度为i的数字，第i位上数字是j的有多少个符合条件的。
int digit[15];
int abs(int a)
{
    if(a<0)return -a;
    else return a;
}
void init()
{
    memset(dp,0,sizeof(dp));
    for(int i=0;i<=9;i++)
    {
        dp[1][i]=1;
    }
    for(int i=2;i<=11;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];
                }
            }
        }
    }
}
int callen(int val)
{
    int num=0;
    while(val)
    {
        val/=10;
        num++;
    }
    return num;
}
void caldig(int val)
{
    memset(digit,0,sizeof(digit));
    int cont=1;
    while(val)
    {
        digit[cont]=val%10;
        val/=10;
        cont++;
    }
}
int cal(int val)
{
    int len=callen(val);
    caldig(val);
    int sum=0;
    for(int i=1;i<len;i++)
    {
        for(int j=1;j<=9;j++)
        {
            sum+=dp[i][j];
        }
    }
    for(int j=1;j<digit[len];j++)
    {
        sum+=dp[len][j];
    }
    for(int i=len-1;i>=1;i--)
    {
        for(int j=0;j<digit[i];j++)
        {
            if(abs(j-digit[i+1])>=2)
            sum+=dp[i][j];
        }
        if(abs(digit[i]-digit[i+1])<2)break;
    }
    return sum;
}
int main()
{
    init();
    int a,b;
    while(~scanf("%d%d",&a,&b))
    {
       printf("%d\n",cal(b+1)-cal(a));
    }
}