[校内模拟][数论] NOIP Practice T1 数论谜题

T1

题意

求
$$\sum _{l,r}\omega (\prod _{i\in [l,r]}{a}_i)$$
其中$\omega (x)$为$x$的质因子个数

正解

转化/换个角度

每个区间某个质数第一次出现时，对区间有一个贡献

#include<bits/stdc++.h>
using namespace std;
#define in Read()
int in{
	int i=0,f=1;char ch=0;
	while(!isdigit(ch)&&ch!='-') ch=getchar();
	if(ch=='-') ch=getchar(),f=-1;
	while(isdigit(ch)) i=(i<<3)+(i<<1)+ch-48,ch=getchar();
	return i*f;
}

typedef long long ll;
const int N=1e6+5;
int n,a[N],vis[N],prime[N],cnt,pre[N];
ll ans;

void sieve(int n){
	for(int i=2;i<=n;++i){
		if(!vis[i]){
			vis[i]=i;
			prime[++cnt]=i;
		}
		for(int j=1;j<=cnt;++j){
			if(prime[j]>vis[i]||prime[j]>n/i) break;
			vis[i*prime[j]]=prime[j];
		}
	}
}


int main(){
	n=in;
	sieve(1e6);
	for(int i=1;i<=n;++i){
		int x=in;
		while(x>1){
			int p=vis[x];
			while(!(x%p)) x/=p;
			ans+=1ll*(n-i+1)*(i-pre[p]);
			pre[p]=i;
		}
	}
	cout<<ans;
	return 0;
}