【BZOJ1026】数位DP

最新推荐文章于 2020-10-03 21:30:32 发布

lyx_2016

最新推荐文章于 2020-10-03 21:30:32 发布

阅读量263

点赞数

分类专栏：题目文章标签： dp

本文链接：https://blog.csdn.net/lyx_2016/article/details/68484296

版权

题目专栏收录该内容

81 篇文章 0 订阅

订阅专栏

dp[i][j]表示i位数，首位为j的符合要求的数有多少个，嗯和不要62那道题的思路差不多。

处理n的时候，因为它要求没有前导0，这个我想了好久，如果再开一维状态记录它前面有没有非0数的话我不是很会写。所以我们先把比n位数小的，比如说1~9中符合要求的，加在答案里，然后是10~99,100~999，这样。然后再把n那个位数级别的，直到n那一位上数的上一个，加到答案中。这些循环都是从1开始的，避免了前导0。

举个栗子，679。先把1~9、10~99的加上，再加上100~500。然后我们看60x，6 - 0 > 2，可以，加上0x【就是dp[2][0]】；61x，可以，加上1x……直到65x，发现不满足了，那它往后一定也是不满足的，它的后几位就直接不用扫了，跳出循环。
注意一下0和1的问题就可以了。

#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;

const int LEN = 11;

int dp[LEN][10], d[LEN];

void prepare()
{
    //dp[0][0] = 1;
    for (int i = 0; i <= 9; i++)
    {
        dp[1][i] = 1;
    }
    for (int i = 1; i <= LEN; i++)
    {
        for (int j = 0; j <= 9; j++)
        {
            //dp[i][j] += dp[i][j - 1];
            for (int k = 0; k <= 9; k++)
            {
                if (abs(j - k) >= 2)
                {
                    dp[i][j] += dp[i - 1][k];
                }
            }
            //printf("%d %d %d\n", i, j, dp[i][j]);
        }
    }
}


int solve(int n)
{
    int len = 0;
    while (n)
    {
        len++;
        d[len] = n % 10;
        n /= 10;
    }

    int ans = 0;

    for (int i = 1; i < len; i++)
    {
        for (int j = 1; j <= 9; j++)
        {
            ans += dp[i][j];
        }
    }
    for (int j = 1; j < d[len]; j++)
    {
        ans += dp[len][j];
    }
    // if (len == 1 && d[len] < 2)
    // {
    //     for (int j = 0; j < d[len]; j++)
    //     {
    //         ans += dp[len][j];
    //     }
    // }

    for (int i = len; i >= 2; i--)
    {
        for (int j = 0; j < d[i - 1]; j++)
        {
            if (abs(d[i] - j) >= 2)
            {
                ans += dp[i - 1][j];
            }
        }
        if (abs(d[i] - d[i - 1]) < 2) break;
    }

    // for (int i = 1; i <= len; i++)
    // {
    //     ans -= dp[i][0];
    // }
    return ans;
}

int main()
{
    prepare();
    int a, b;
    scanf("%d %d", &a, &b);
    //printf("%d %d\n", solve(b + 1), solve(a));
    printf("%d\n", solve(b + 1) - solve(a));
    return 0;
}