计数问题~

#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);//测试目标数非零的测试数据
}

复写代码

#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=20;
int l,r,len,a[N],dp[N][N];
int dfs(int pos,int sum,int num,int lead,int limit)
{
    if(!pos)
    {
        if(lead&&!num)return 1;
        return sum;
    }
    if(!limit&&!lead&&dp[pos][sum]!=-1)return dp[pos][sum];
    int res=0,up=limit?a[pos]:9;
    for(int i=0;i<=up;i++)
    {
        int t;
        if(i==num)
        {
            if(!num)t=sum+(lead==0);
            else t=sum+1;
        }
        else t=sum;
        res+=dfs(pos-1,t,num,lead&&i==0,limit&&i==up);
    }
    return limit||lead?res:dp[pos][sum]=res;
}
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);
}
int main()
{
    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;
    }
    return 0;
}