题意
有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。