C. Add One(dp)

https://codeforces.com/contest/1513/problem/C

---

思路：

第一发tle。因为没看到他是t组也算进去的。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+1000;
typedef long long LL;
const LL mod=1e9+7;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
map<LL,LL>map1;
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   LL t;cin>>t;
   while(t--){
      map1.clear();
      string s;cin>>s;
      LL n=s.size();
      LL m;cin>>m;
      for(LL i=0;i<s.size();i++){
          map1[(s[i]-'0')]++;
      }
      for(LL i=1;i<=m;i++){
          map<LL,LL>map2;
          for(auto j:map1){
              LL temp=j.first;
              temp++;
              if(temp==10){
                 map2[1]=(map2[1]%mod+j.second%mod)%mod;
                 map2[0]=(map2[0]%mod+j.second%mod)%mod;
              }
              else{
                 map2[temp]=(map2[temp]%mod+j.second%mod)%mod;;
              }
          }
      }
      LL ans=0;
      for(auto i:map1){
         ans=(ans%mod+i.second%mod)%mod;
      }
      cout<<ans<<"\n";
   }
   return 0;
}

那么多组，是否可以预处理呢?

如果每个数组都预处理，1e9个数字肯定不够存。那么按位进行考虑，预处理出初始状态是一位0的m次操作后的总长度。

对于答案，其每一位数字如果不是0，那么可以看成其数字就是给0多做了几次操作，答案就是res[n%10+m]。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+1000;
typedef long long LL;
const LL mod=1e9+7;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL dp[maxn][12];
LL ans[maxn];
void pre(){
    dp[0][0]=1;
    ans[0]=1;
    for(LL i=1;i<maxn;i++){
        for(LL j=0;j<=9;j++){
            if(j==0){
                dp[i][j]=(dp[i-1][9]%mod);
            }
            else if(j==1){
                dp[i][j]=(dp[i-1][9]%mod+dp[i-1][0]%mod)%mod;
            }
            else{
                dp[i][j]=(dp[i-1][j-1]%mod);
            }
            ans[i]=(ans[i]%mod+dp[i][j]%mod)%mod;
        }
    }
}
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   pre();
   LL t;cin>>t;
   while(t--){
      LL n,m;cin>>n>>m;
      LL res=0;
      while(n){
          res=(res%mod+ans[n%10+m]%mod)%mod;
          n/=10;
      }
      cout<<res<<"\n";
   }
   return 0;
}