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;
}