J - Prime Game（数论分解质因子)

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=1e6+100;
vector<int>v[maxn];
int prime[maxn];
bool biaoji[maxn];
int a[maxn]; 
int cnt=0;
void get(){
    for(int i=2;i<maxn;i++){
        if(!biaoji[i]) prime[++cnt]=i;
        for(int j=1;j<=cnt&&i*prime[j]<maxn;j++){
            biaoji[i*prime[j]]=1;
            if(i%prime[j]==0){
                break;
            }
        }
    } 
}
void di(int x){
    int t=a[x];
    for(int j=1;j<=cnt&&prime[j]*prime[j]<=t;j++){
        if(t%prime[j]==0){
            v[prime[j]].push_back(x);
            while(t%prime[j]==0){
                t/=prime[j];
            }
        }
    }
    if(t!=1){
        v[t].push_back(x);
    } 
} 
int main(){
    get();
    int n;
    cin>>n;
    for(int i=2;i<=maxn;i++){
        v[i].push_back(0);
    }
    for(int i=1;i<=n;i++){
        cin>>a[i];
        di(i);
    } 
    ll ans=0;
    for(int i=1;i<=cnt;i++){
        for(int j=1;j<v[prime[i]].size();j++){
            ans=ans+1ll*(v[prime[i]][j]-v[prime[i]][j-1])*1ll*(n-v[prime[i]][j]+1); 
        } 
    }
    cout<<ans<<endl;
}