【数位统计DP】计数问题

最新推荐文章于 2024-10-02 00:00:31 发布

Nathan Qian

最新推荐文章于 2024-10-02 00:00:31 发布

阅读量95

点赞数

分类专栏：蓝桥刷题冲冲！文章标签：蓝桥杯 c++ 拓扑学

本文链接：https://blog.csdn.net/nathanqian123/article/details/122958669

版权

蓝桥刷题冲冲！专栏收录该内容

58 篇文章 1 订阅

订阅专栏

该博客主要介绍了如何使用C++解决AcWing平台上的338号计数问题。内容涉及从数字串中统计特定数字出现的次数，通过前缀和的方法进行区间计算，并给出了完整的代码实现，包括get_num和count两个关键函数。

摘要由CSDN通过智能技术生成

题目

338. 计数问题 - AcWing题库

AcWing 338. 计数问题 - AcWing

解释

count函数来求从(1,num)数字（0-9）出现的次数
如果要求区间(a,b)，则相当于从前缀和count中作差求结果
对于num=abcdefg(如下图考虑）注意x=0时的特殊情况

代码段

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int get_num(vector<int>num,int l,int r)
{
    int n=0;
    for(int i=r;i>=l;i--)
    n=n*10+num[i];
    return n;
}
int power10(int x)
{
    int ans=1;
    while(x--)
    {
        ans*=10;
    }
        
        return ans;
}
int count(int n,int x)
{
    vector<int>num;
    while(n)
    {
        num.push_back(n%10);
        n/=10;
    }
    n =num.size();
    int ans=0;
    for(int i=n-1-!x;i>=0;i--)
    {
        if(i<n-1)
        {
            ans+=get_num(num,i+1,n-1)*power10(i);
            if(!x)ans-=power10(i);
            
        }
        if(num[i]==x)ans+=get_num(num,0,i-1)+1;
        else if(num[i]>x)ans+=power10(i);
    }
    return ans;
}
int main()
{
    int a,b;
    while(cin>>a>>b,a)
    {
        if(a>b)
        swap(a,b);
        for(int i=0;i<10;i++)
        cout<<count(b,i)-count(a-1,i)<<' ';
        cout<<endl;
    }
    return 0;
}