CF 920F SUM and REPLACE(线性递推&线段树)
题目大意
设以函数 D ( i ) D(i) D(i)为i的因数的数量。现在给出一段序列{a}。有两种操作,操作1为将l和r之间的所有的数 a i a_i ai变为 D ( a i ) D(a_i) D(ai),另一种操作求l,r之间 a i a_i ai的和
解题思路
首先线性地递推出 所有的 D ( i ) D(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));
}
}
}