思路:
区间修改线段树。但是这一题不能使用懒惰标记,因为需要精确修改区间内所有点,然后向上更新。但是为了降低时间消耗,减少不必要的更新操作,我们引入了一个ok数组。
当d[i]等于1或2时就没有必要进行d操作了,ok[i]=1.
当区间内的所有点都是1或2时,那么这个区间就不用更新了。ok数组向上更新即可。
d数组可以用筛法求。第一层循环枚举因子,第二层循环枚举因子的倍数。
代码:
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<iostream>
#include<queue>
using namespace std;
int n,m;
int d[1000005],ok[4*300005];
long long sum[4*300005];
void init()
{
memset(sum,0,sizeof sum);
memset(ok,0,sizeof ok);
for(int i=1;i<=1000000;i++)
{
for(int j=i;j<=1000000;j+=i)
d[j]++;
}
}
void build(int v,int l,int r)
{
if(l==r)
{
long long x;
scanf("%lld",&sum[v]);
if(sum[v]==1 || sum[v]==2) ok[v]=1;
return ;
}
int mid=(l+r)/2;
build(v*2,l,mid);
build(v*2+1,mid+1,r);
sum[v]=sum[v*2]+sum[v*2+1];
ok[v]=ok[v*2] & ok[v*2+1];
}
void updata(int v,int l,int r,int ql,int qr)
{
if(ok[v]) return ;
if(l==r)
{
sum[v]=d[sum[v]];
if(sum[v]==1 || sum[v]==2) ok[v]=1;
return ;
}
int mid=(l+r)/2;
if(ql<=mid) updata(v*2,l,mid,ql,qr);
if(qr>mid) updata(v*2+1,mid+1,r,ql,qr);
sum[v]=sum[v*2]+sum[v*2+1];
ok[v]=ok[v*2] & ok[v*2+1];
}
long long query(int v,int l,int r,int ql,int qr)
{
if(ql<=l && r<=qr)
return sum[v];
int mid=(l+r)/2;
long long s=0;
if(ql<=mid) s+=query(v*2,l,mid,ql,qr);
if(qr>mid) s+=query(v*2+1,mid+1,r,ql,qr);
return s;
}
int main()
{
scanf("%d%d",&n,&m);
init();
build(1,1,n);
int op,l,r;
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&op,&l,&r);
if(op==1) updata(1,1,n,l,r);
else printf("%lld\n",query(1,1,n,l,r));
}
return 0;
}