C. Add One(预处理)

题意
有T组询问, 1 < = T < = 2 e 5 1<=T<=2e5 1<=T<=2e5
然后每组输入n和m, 1 < = n < = 1 e 9 , 1 < = m < = 2 e 5 1<=n<=1e9,1<=m<=2e5 1<=n<=1e9,1<=m<=2e5
我们每次对n的每一位加1,加m次,问最后n有多少位。
思路
我昨天晚上写这个题的时候一直以为T组询问的m的总和是2e5.
所以我就在想o(n)的做法。
我首先我把n的每一位用桶存起来,然后就跑m次循环,每次循环就对1到9进行转移。
暴力代码

#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
map<int,int>mp;
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        mp.clear();
        string s;
        int m;
        cin>>s>>m;
        for(int i=0 ; i<s.length() ; i++)
        {
            mp[s[i]-'0']++;
        }
        for(int i=1 ; i<=m ; i++)
        {
            int cnt=0;
            for(int j=9 ; j>=0 ; j--)
            {
                if(mp[j]==0) continue;
                if(j!=9)
                {
                    mp[j]%=mod;
                    mp[j+1]+=mp[j];
                    mp[j+1]%=mod;
                    mp[j]=0;
                }
                else
                {
                    cnt=mp[9];
                    mp[9]=0;
                    cnt%=mod;
                }
            }
            mp[0]+=cnt;
            mp[0]%=mod;
            mp[1]+=cnt;
            mp[1]%=mod;
          //  for(int j=0 ; j<=9 ; j++) cout<<mp[j]<<" ";
          //  cout<<endl;
        }
        ll ans=0;
        for(int i=0 ; i<=9 ; i++)
        {
            ans=ans+mp[i];
            ans%=mod;
        }
        cout<<ans<<endl;
    }
    return 0;
}

但是交的时候却T了,一直以为是自己的那里写法有问题,一直在修改,但是一直交到最后都是超时。
今天早上看题的时候才发现,每组询问的m的数量都可能是2e5,所以昨天晚上那样肯定是不行的。
后来我发现了一种性质,那就是n的每一位都互相没有任何关系,我们是可以预处理出0~9的每个数的加m次的结果,最后只需要看n的位数,来把所有位数的数加在一起。
最后写出来复杂度应该在1e7到1e8左右,所有我们读入和输出的时候应该用scanf和printf,如果用cin的话会超时的。
代码

#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
ll vis[10];
ll dp[10][N];
void init()
{
    for(int k=0 ; k<=9 ; k++)
    {
        for(int i=0 ; i<=9 ; i++) vis[i]=0;
        vis[k]++;
        for(int i=1 ; i<=200005 ; i++)
        {
            int cnt=0;
            for(int j=9 ; j>=0 ; j--)
            {
                if(vis[j]==0) continue;
                if(j!=9)
                {
                    vis[j]%=mod;
                    vis[j+1]+=vis[j];
                    vis[j+1]%=mod;
                    vis[j]=0;
                }
                else
                {
                    cnt=vis[9];
                    vis[9]=0;
                    cnt%=mod;
                }
            }
            vis[0]+=cnt;
            vis[0]%=mod;
            vis[1]+=cnt;
            vis[1]%=mod;
            for(int z=0 ; z<=9 ; z++)
            {
                dp[k][i]+=vis[z];
                dp[k][i]%=mod;
            }
        }
    }
}
int main()
{
    init();
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        ll ans=0;
        while(n)
        {
            ans=ans+dp[n%10][m];
            ans%=mod;
            n/=10;
        }
        printf("%lld\n",ans);
    }
    return 0;
}

总结
写题的时候没有看清楚题的范围,而且没有发现重要的性质,还有看到有好多大佬写的是DP。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值