2018南京区域赛——J – Prime Game

Gym – 101981J
数论
素数筛，计算因子贡献值
找因子的时候只用找到sqrt(n),除剩下的为最后一个素数因子

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
int vis[maxn];
int prime[maxn],cnt=0;
int maxin;
void isprime() {
    memset(vis,0,sizeof(vis));
    for(int i=2;i<=maxin;i++) {
        if(!vis[i]) prime[cnt++]=i;
        for(int j=0;j<cnt&&i*prime[j]<=maxin;j++) {
            vis[i*prime[j]]=1;
            if(i%prime[j]==0)
                break;
        }
    }
}
int in[maxn],pre[maxn];
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    memset(pre,0,sizeof(pre));
    int n; cin>>n;
    maxin=0;
    for(int i=0;i<n;i++) {
        cin>>in[i];
        if(in[i]>maxin)
            maxin=in[i];
    }
    isprime();
    long long ans=0;
    for(int i=0;i<n;i++) {
        for(int j=0;j<cnt&&in[i]>=prime[j]*prime[j];j++) {
            if(in[i]%prime[j]==0) {
                long long l=i+1-pre[prime[j]];
                long long r=n-i;
                ans+=l*r;
                pre[prime[j]]=i+1;
                while(in[i]%prime[j]==0) {
                    in[i]/=prime[j];
                }
            }
        }
        if(in[i]!=1) {
            long long l=i+1-pre[in[i]];
            long long r=n-i;
            ans+=l*r;
            pre[in[i]]=i+1;
        }
    }
    cout<<ans<<endl;
}