计数问题（数位DP）

原题链接：计数问题

思路：

采取DFS方法进行搜索，设置变量f[i][j]，表示搜索到第i位时目标数已经出现了j次的结果，len表示有多少位，pos表示枚举到哪一位，limit表示前几位有没有贴近上界的限制，lead表示有没有前导零，sum表示前几位目标数出现了几次，num表示目标数。

//毕老师的注释
#include <bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
#define ll long long
#define Mirai ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
#define COUT cout << "###" << endl;
#define pb push_back
using namespace std;
typedef pair<int, int> PII;
const int N = 15;
int l, r, len, a[N], dp[N][N];
//dp[i][j]的意义是在枚举到pos位并且在此之前目标数已经出现j次的结果数
//目标数在此之前具体出现在那个位置并不影响当前的方案数(只要当前没有限制)
int dfs(int pos, int sum, int num, int lead, int limit)//参数依次为：当前枚举数位，在此之前目标数的出现次数，目标数，是否为前导零状态，取值限制
//lead为1判断为前导零状态，一直枚举到某个数位的数值不取0的情况，lead就会变为0并且一直保持到最后一位
//lead主要用来处理特判目标数为0时的前导零问题
{
    if (!pos)//边界条件，枚举到最后一位
    {
        if (lead && !num)//如果枚举到最后一位仍然为前导零状态，并且要查找的数为零，则说明目前枚举的数是零，则返回1(0只有一个0)
            return 1;
        return sum;//否则返回此前目标数出现的次数
    }
    if (!limit && !lead && dp[pos][sum] != -1)//如果无限制，且无前导零(记忆化搜索直接返回结果)
        return dp[pos][sum];
    int res = 0, up = limit ? a[pos] : 9;//up为取值上界
    for (int i = 0; i <= up; i++)//枚举当前可以选的所有数(下界为0，上界为up)
    {
        int t;//t用来存储到当前位置目标数出现的次数
        if (i == num)//如果枚举到目标数
        {
            if (!num)//如果目标数为零
            {
                t = sum + (lead == 0);//只有lead不为零也就是当前数位不为前导零的情况下sum才会加一，当前的零才会被计入
            }
            else
            {
                t = sum + 1;//如果不为零直接加1
            }
        }
        else
        {
            t = sum;//如果当前不选择目标数，则sum不变
        }
        //cout<<pos<<" "<<t<<" "<<sum<<" "<<i<<" "<<" "<<limit<<" "<<dp[1][0]<<" "<<lead<<"  "<<res<<endl;//测试输出
        res += dfs(pos - 1, t, num, lead && i == 0, limit && i == up);
        //举例数字318
        //第一次枚举到最后一位(上界为3)的时候，首先枚举0，边界条件范围0
        //枚举2，此时t=1，dfs边界时返回sum=1，
        //此后枚举其他不为2的数都返回0，由于此时limit=0无限制，也就是在枚举到最后一位，且在枚举最后一位之前2出现过0次的结果可以保存下来(pos=1,sum=0)
        //修正：此时第一次枚举完最后一位时dp[1][0]并没有被保存下来，因为此时lead为1，不能保存
        //需要第二次进入也就是倒数第二位取到1时
        //这样做可以防止枚举零时出现统计错误
        //原因解释：枚举最高位时，最开始先枚举0，然而此时为前导零状态，此分支返回时，如果此时保存了结果，即为dp[3][0]=10
        //意义为枚举最高位且在此之前没有出现目标数字零的情况一共有10种
        //分别为0，10，20，30，40，50，60，70，80，90中的零，而如果此种情况保存了下来
        //最高位再枚举1的时候，此时不为前导零状态，而这种状态同样也为dp[3][0]，即直接调用了已经保存的结果10
        //然而这个结果对于最高位不为零的情况并不适用，因为最高位不为零，就需要计算不为最高位的其他位上的零，
        //例如103，100，108等等,所以前导零状态下保存的目标数为零的结果是不合法的
            //以下为参数解释//
        //pos-1：dfs下一位
        //t：num(目标数)已经出现的次数(更新过后的sum)
        //num:目标数
        //lead&&i==0：lead最开始为1，如果当前数为零则说明仍然是前导零，则lead仍然为1，若枚举到不是零，则说明在当前路径下从此处开始不为前导零
        //limit&&i==up判断下一位的dfs有无限制的套路
    }
    //cout<<pos<<" "<<num<<" "<<sum<<" "<<endl;//测试输出
    // return limit?res:dp[pos][sum]=res;//此种返回方式不会影响除了0之外其他目标数的统计，但统计零会导致结果不正确
    return limit ? res : (lead ? res : dp[pos][sum] = res);//必须保证limit==0&&lead==0才能保存结果
}
int cal(int x, int num)
{
    memset(dp, -1, sizeof dp);
    len = 0;
    while (x)
        a[++len] = x % 10, x /= 10;
    return dfs(len, 0, num, 1, 1);
}
signed main()
{
    while (cin >> l >> r, l || r)//l和r必须有一个不为零，只要有一个为真结果即为真，也就是两者都为零的情况退出循环
    {
        if (l > r)//特判不合法情况
            swap(l, r);
        for (int i = 0; i <= 9; i++)//输出从0到9的个数
            cout << cal(r, i) - cal(l - 1, i) << " ";
        cout << endl;
    }
    // cout<<cal(318,0);//测试目标数为零的测试数据
    //cout<<cal(318,2);//测试目标数非零的测试数据
}

复写

#define _CRT_SECURE_NO_WARNINGS
#include<bits/stdc++.h>
using namespace std;
#define enl , writestr("\n")
#define int long long
#define endl "\n"
#define PII pair<int, int>
#define ULL unsigned long long
#define xx first
#define yy second
#define PI acos(-1)
const int P = 131;//131,13331
const int N = 2e5 + 10;
const int M = 1e3 + 10;
const int INF = 1e18;
const int ENF = -1e18;
const int mod = 1e9 + 7;//998244353
const int modd = 998244353;
void ClearFloat() { ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0); }int read() { int ret = 0, f = 1; char ch = getchar(); while ('0' > ch || ch > '9') { if (ch == '-')f = -1; ch = getchar(); }while ('0' <= ch && ch <= '9') { ret = ret * 10 + ch - '0'; ch = getchar(); }return ret * f; }void write(int x) { char num[30]; int cnt = 0; if (x == 0) { putchar('0'); return; }if (x < 0) { putchar('-'); x = -x; }while (x > 0) { num[cnt++] = x % 10 + '0'; x /= 10; }while (cnt > 0) { putchar(num[--cnt]); } }double readdou() { double x = 0; int flag = 0; char ch = 0; while (!isdigit(ch)) { flag |= (ch == '-'); ch = getchar(); } while (isdigit(ch)) { x = x * 10 + (ch - '0'); ch = getchar(); }if (ch != '.') return flag ? -x : x; int f = 1; ch = getchar(); while (isdigit(ch)) { x = x + (ch - '0') * pow(10, -f); f++; ch = getchar(); }return flag ? -x : x; }void wridou(int x) { char num[30]; int cnt = 0; if (x == 0) { putchar('0'); return; }if (x < 0) { putchar('-'); x = -x; }while (x > 0) { num[cnt++] = x % 10 + '0'; x /= 10; }while (cnt > 0) { putchar(num[--cnt]); } }int gcd(int a, int b) { if (b) while ((a %= b) && (b %= a)); return a + b; }int lcm(int a, int b) { return a * b / gcd(a, b); }
int n, m, k, f[15][15], l, r, a[N], len;
int dfs(int pos, int sum, int num, int limit, int lead)
{
	if (!pos)
	{
		if (lead && !num) return 1;
		return sum;
	}
	if (!limit && !lead && ~f[pos][sum]) return f[pos][sum];
	int res = 0, up = limit ? a[pos] : 9;
	for (int i = 0; i <= up; i++)
	{
		int t = 0;
		if (i == num)
		{
			if (!num) t = sum + (lead == 0);
			else t = sum + 1;
		}
		else t = sum;
		res += dfs(pos - 1, t, num, limit && i == up, lead && i == 0);
	}
	return limit ? res : (lead ? res : f[pos][sum] = res);
}
int cal(int x, int num)
{
	memset(f, -1, sizeof f);
	len = 0;
	while (x > 0) a[++len] = x % 10, x /= 10;
	return dfs(len, 0, num, 1, 1);
}
signed main()
{
	ClearFloat();
	while (cin >> l >> r, l || r)
	{
		if (l > r) swap(l, r);
		for (int i = 0; i <= 9; i++) cout << cal(r, i) - cal(l - 1, i) << ' ';
		cout << endl;
	}
}