CF 920F SUM and REPLACE(线性递推&线段树)

CF 920F SUM and REPLACE(线性递推&线段树)

题目大意

设以函数$D(i)$为i的因数的数量。现在给出一段序列{a}。有两种操作，操作1为将l和r之间的所有的数$a_i$变为$D(a_i)$，另一种操作求l,r之间$a_i$的和

解题思路

首先线性地递推出 所有的$D(i)$，用一棵线段树维护这一段序列，每次更新一段序列时暴力更新，据打表最多6次更新，其后更新就不会产生变化，因此每次更新时如果这一段的最大值都已经小于等于2则这一段更新对数值不会缠任何的影响，因此无需更新，这样最后就会产生序列长度*6次数的更新

AC代码

#include<bits/stdc++.h>
using namespace std;
#define int long long 
const int maxn=1e6+5;
const int maxm=3e5+5;
int sigma[maxn];
bool prime[maxn];
int p[maxn];
int tot=0;
int a[maxm];
inline int rson(int x){return x<<1|1;}
inline int lson(int x){return x<<1;}
inline int md(int l,int r){return (l+r)>>1;}
struct node{
	int l,r,mx;
	int sum;
}seg[maxm<<2];
void init()
{
	sigma[1]=1;
	for(int i=2;i<maxn;i++) prime[i]=true;
	for(int i=2;i<maxn;i++)
	{
		if(prime[i]==true)
		{
			p[tot++]=i;
			sigma[i]=2;
		}
		for(int j=0;j<tot&&p[j]*i<maxn;j++)
		{
			prime[i*p[j]]=false;
			if(i%p[j]==0)
			{
				sigma[i*p[j]]=sigma[i]*2-sigma[i/p[j]];
				break;
			}
			else
			{
				sigma[i*p[j]]=2*sigma[i];
			}
		}
	}
}
void build(int k,int l,int r)
{
	seg[k].l=l,seg[k].r=r;
	if(l==r)
	{
		seg[k].mx=seg[k].sum=a[l];
		return ;
	}
	int mid=md(l,r);
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	seg[k].sum=seg[k<<1].sum+seg[k<<1|1].sum;
	seg[k].mx=max(seg[k<<1].mx,seg[k<<1|1].mx);
}
void update(int k,int l,int r)
{
	if(seg[k].mx<=2) return ;
	if(seg[k].l==seg[k].r)
	{
		seg[k].mx=seg[k].sum=sigma[seg[k].mx];
		return ;
	}
	int mid=md(seg[k].l,seg[k].r);
	if(mid>=l) update(k<<1,l,r);
	if(mid<r) update(k<<1|1,l,r);
	seg[k].sum=seg[k<<1].sum+seg[k<<1|1].sum;
	seg[k].mx=max(seg[k<<1].mx,seg[k<<1|1].mx);
}
int query(int k,int l,int r)
{
	if(seg[k].l>=l&&seg[k].r<=r)
	{
		return seg[k].sum;
	}
	int mid=md(seg[k].l,seg[k].r);
	int ans=0;
	if(mid>=l) ans+=query(k<<1,l,r);
	if(mid<r) ans+=query(k<<1|1,l,r);
	return ans;
}
int32_t main()
{
	int n,m;
	init();
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
	}
	build(1,1,n);
	int l,r,op;
	for(int i=1;i<=m;i++)
	{
		scanf("%lld%lld%lld",&op,&l,&r);
		if(op==1)//update
		{
			update(1,l,r);
		}
		else //query
		{
			printf("%lld\n",query(1,l,r));
		}
	}
}