【CDOJ 250】windy数

【CDOJ 250】windy数

数位dp 限制条件是相邻两个数差至少为2 预处理dp数组 dp[i][j]表示i为高位且i位为数字j时满足题意的种数

代码如下:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>

using namespace std;

int dp[11][10];

void Init()
{
    memset(dp,0,sizeof(dp));
    int i,j,k;
    for(i = 0; i <= 9; ++i) dp[1][i]++;

    for(i = 2; i <= 10; ++i)
        for(j = 0; j <= 9; ++j)
            for(k = 0; k <= 9; ++k)
                if(abs(j-k) >= 2) dp[i][j] += dp[i-1][k];
}

int Solve(int n)
{
    int ans = 0,len = 0,bit[12],i,j;
    while(n)
    {
        bit[++len] = n%10;
        n /= 10;
    }

    for(i = len-1; i; --i)//1~len-1位可取任意数
        for(j = 1; j <= 9; ++j)
            ans += dp[i][j];

    for(i = 1; i < bit[len]; ++i)//len位<该位数字时的满足种数
        ans += dp[len][i];

    for(i = len-1; i; --i)//高->低进位
    {
        for(j = 0; j < bit[i]; ++j)
            if(abs(j-bit[i+1]) >= 2)
                ans += dp[i][j];

        if(abs(bit[i]-bit[i+1])<2)//出现不满足的前缀跳出
            break;
    }
    return ans;
}

int main()
{
    int i,j;
    Init();
    int a,b;
    while(~scanf("%d%d",&a,&b))
    {
        printf("%d\n",Solve(b+1)-Solve(a));
    }
    return 0;
}