2021牛客寒假算法基础集训营2 I.牛牛的“质因数”（预处理筛）

2021牛客寒假算法基础集训营2 I.牛牛的“质因数”（预处理筛）

题目链接：https://ac.nowcoder.com/acm/contest/9982/I（付费比赛）
题面：
题解：对于函数 $F(x)$ 的值，显然只与 $x$ 的质因数有关。不难联想到使用欧拉筛筛出 $F(x)$ ，因为对于每个正整数，欧拉筛都能找出它们的最小质因数。再考虑最小质因数一定是拼接在最前面的，那么还需要一个数组 $sz$ 维护 $F(x)$ 原本值所具有的长度，我们即可以得到 $F（i\ast j）=F（i）+j\ast 10^{sz[i]}$ ( $j$ 为 $i\ast j$ 的最小质因数) 的转移式，同时对于每个质数 $p$ ，有 $F(p)=p$ 。
下面为上述思路得出的AC代码。

#include<bits/stdc++.h>
using namespace std;
#define debug(x) cerr<<#x<<":"<<x<<endl
#define debug2(x,y) cerr<<#x<<":"<<x<<"  "<<#y<<":"<<y<<endl
#define debug3(x,y,z) cerr<<#x<<":"<<x<<"  "<<#y<<":"<<y<<"  "<<#z<<":"<<z<<endl
typedef long long ll;
struct fastio{
    fastio(){
        ios::sync_with_stdio(0);
        cin.tie(0);
        cout.tie(0);
    }
}fio;
const int M=1e9+7;
int N;
vector<int> prime,sum,res,sz;
ll qpow(ll x,ll y){
    ll res=1;
    while(y){
        if(y&1) res=res*x%M;
        x=x*x%M;
        y>>=1;
    }
    return res;
}
void init(){
    sum=res=sz=vector<int>(N,0);
    for(int i=2;i<N;++i){
        if(!res[i]){
            prime.push_back(i);
            res[i]=i;
            sz[i]=log(i)/log(10)+1;
        }
        for(auto j:prime){
            if(i*j>=N) break;
            res[i*j]=((ll)res[i]+j*qpow(10,sz[i]))%M;
            sz[i*j]=sz[i]+sz[j];
            if(i%j==0) break;
        }
        sum[i]=((ll)res[i]+sum[i-1])%M;
    }
}
signed main(){
    int n;
    cin>>n;
    N=n+5;
    init();
    cout<<sum[n]<<endl;
}

这种做法的复杂度大概是 $O(nlogw)$ ，这是因为我们处理合数情况下的 $F(x)$ 需要计算一次10的幂。考虑进一步优化，其实 $sz$ 数组的最大值并不会很大，因为 $sz$ 储放的是一个数的十进制长度。那么我们同样也可以预处理10的幂的值，把复杂度降为 $O(n)$ （大概吧）。

#include<bits/stdc++.h>
using namespace std;
#define debug(x) cerr<<#x<<":"<<x<<endl
#define debug2(x,y) cerr<<#x<<":"<<x<<"  "<<#y<<":"<<y<<endl
#define debug3(x,y,z) cerr<<#x<<":"<<x<<"  "<<#y<<":"<<y<<"  "<<#z<<":"<<z<<endl
typedef long long ll;
struct fastio{
    fastio(){
        ios::sync_with_stdio(0);
        cin.tie(0);
        cout.tie(0);
    }
}fio;
const int M=1e9+7;
int N;
vector<int> prime,sum,res,sz,memo;
void init(){
    memo=sum=res=sz=vector<int>(N,0);
    memo[1]=10;
    for(int i=2;i<N;++i){
        memo[i]=(ll)memo[i-1]*10%M;
        if(!res[i]){
            prime.push_back(i);
            res[i]=i;
            sz[i]=log(i)/log(10)+1;
        }
        for(auto j:prime){
            if(i*j>=N) break;
            res[i*j]=((ll)res[i]+j*memo[sz[i]])%M;
            sz[i*j]=sz[i]+sz[j];
            if(i%j==0) break;
        }
        sum[i]=((ll)res[i]+sum[i-1])%M;
    }
}
signed main(){
    int n;
    cin>>n;
    N=n+5;
    init();
    cout<<sum[n]<<endl;
}

上面是“优化”后的代码，由下图可见常数还是占了挺大比例的，并没有优化到什么。。。