数位动态规划：数字游戏

题目描述

科协里最近很流行数字游戏。

某人命名了一种不降数，这种数字必须满足从左到右各位数字呈非下降关系，如 123，446。

现在大家决定玩一个游戏，指定一个整数闭区间 $[a,b]$，问这个区间内有多少个不降数。

输入格式

输入包含多组测试数据。

每组数据占一行，包含两个整数 $a$ 和 $b$。

输出格式

每行给出一组测试数据的答案，即 $[a,b]$ 之间有多少不降数。

数据范围

$1\le a\le b\le 2^{31}-1$

输入样例

1 9
1 19

输出样例

9
18

算法思想

根据题目描述，求区间内数位符合从左到右呈非下降关系的数的个数，可以用数位DP的思想进行分析，例如求区间 $[a,b]$中符合条件的数的个数：

1. 实现函数dp(n)求区间 $[1,n]$中符合条件的数的个数
2. dp(b) - dp(a - 1)求区间 $[a,b]$中符合条件的数的个数

dp(n)可以用分类谈论的方法进行计算。在不能超过n的情况下，从最高位开始向下枚举数n的每一位a[i]，计算出区间$[1-n]$所有符合条件的方案。

如果从集合划分的角度进行分析，可以将所有分类用一棵二叉树表示出来。如下图所示：

1. 左侧分支表示所有符合条件的数中第i位小于a[i]的情况，即$0-a[i]-1$。此时，剩余i位的数字任意取都不会超过n，因此这一类的方案可以通过预先处理出i位数字且最高位不超过a[i] - 1的方案数，然后进行累加即可。
2. 右侧分支表示所有符合条件的数中第i位等于a[i]的情况，此时继续向下分类讨论即可。
3. 最后一位a[0]单独处理，如果能走到最右侧结点a[0]，说明上面每个结点都是非下降的关系，此时总方案数增加1。

算法实现

1. 预处理出f[i][j]，表示有i位数字且最高位为j的不下降数的个数。状态计算：$f[i][j]=\sum f[i-1][k]$, k ≥ j$
2. 预处理出数字n的每一位a[i]
3. 从高位到低位枚举，累加每一位上符合条件的方案数。注意，在向下枚举的过程中，每一个数位不能低于上一位确定的数，可以使用变量last记录上一位的数字。

代码实现

#include <iostream>
#include <vector>
using namespace std;
const int N = 15;
//f[i][j]表示一共有i位数字，且最高位是j的不降数的个数
int f[N][N];

//预处理f[i][j]
//状态表示：f[i][j]表示一共有i位数字，且最高位是j的不降数的个数
//状态计算：f[i][j] = sum{ f[i - 1][k] }, k ≥ j 
void init()
{
    //只有一位数字
    for(int i = 0; i <= 9; i++) f[1][i] = 1;

    //从两位数字开始枚举
    for(int i = 2; i < N; i++)
        for(int j = 0; j <= 9; j++) //枚举最高位
            for(int k = j; k <= 9; k++) //枚举最高位的下一位
                f[i][j] += f[i - 1][k];
}

int dp(int n)
{
    //n为0，只有一种方案就是0
    if(!n) return 1;    

    vector<int> a;
    while(n) a.push_back(n % 10), n /= 10;
    int res = 0;
    int last = 0; //last表示上一位数字是多少
    for(int i = a.size() - 1; i >= 0; i--)
    {
        int x = a[i];
        for(int j = last; j < x; j++) //累加左侧分支
            res += f[i + 1][j];

        if(x < last) break; //不存在右侧分支

        last = x; //更新当前节点选择的数

        //走到最右侧节点，说明上面每个节点都是非下降的关系
        if(!i) res++; 
    }
    return res;
}


int main(){

    init();

    int a, b;
    while(cin >> a >> b)
    {
        cout << dp(b) - dp(a - 1) << endl;    
    }

    return 0;
}