Codeforces 920F SUM and REPLACE-线段树+欧拉筛

传送门

题意：

n个数a，q个操作，两种操作类型：

1.[l,r]区间中每个数替换为这个数的因数个数

2.区间求和

$q,n<=3e5,a_i<=1e6$

Solution:

值为1或2的数是不需要再进行操作的

对于每个线段树节点所代表的区间，开一个布尔变量v表示这段区间是否全为1或2

可以证明每个数最多被替换log次

总复杂度$O(nlog^2n)$

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N=300010; 
int n,m,x,y,z;
int a[N];
struct tree{
    int l,r;
    long long sum;
    bool v;
}tr[4*N];
bool vis[1000010];
int e[1000010],prim[N],cnt,d[1000010];
void get_prim(int n)
{
    d[1]=1;
    for (int i=2;i<=n;i++)
    {
        if (!vis[i])
        {
            prim[++cnt]=i;
            d[i]=2;
            e[i]=1;
        }
        for (int j=1;j<=cnt&&i*prim[j]<=n;j++)
        {
            vis[prim[j]*i]=1;
            if (i%prim[j]==0)
            {
                d[i*prim[j]]=d[i]/(e[i]+1)*(e[i]+2);
                e[i*prim[j]]=e[i]+1;
                break;
            }
            else
            {
                d[i*prim[j]]=d[i]*d[prim[j]];
                e[i*prim[j]]=1;
            }
        }
    }
}
void update(int i)
{
    tr[i].sum=tr[i<<1].sum+tr[i<<1|1].sum;
    tr[i].v=(tr[i<<1].v&tr[i<<1|1].v);
}
void build(int i,int l,int r)
{
    tr[i].l=l,tr[i].r=r;
    if (l==r) {tr[i].sum=a[l];if (a[l]==1||a[l]==2) tr[i].v=1;return;}
    int mid=l+r>>1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
    update(i);
}
void modify(int i,int l,int r)
{
    if (tr[i].v) return;
    int L=tr[i].l,R=tr[i].r;
    if (l>R||L>r) return;
    if (L==R) {tr[i].sum=d[tr[i].sum];if (tr[i].sum==2) tr[i].v=1;return;}
    modify(i<<1,l,r);
    modify(i<<1|1,l,r);
    update(i);
}
long long query(int i,int l,int r)
{
    int L=tr[i].l,R=tr[i].r;
    if (l>R||L>r) return 0;
    if (l<=L&&R<=r) return tr[i].sum;
    long long ans=0;
    ans+=query(i<<1,l,r);
    ans+=query(i<<1|1,l,r);
    //cout<<ans<<endl;
    return ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    get_prim(1000000);
    build(1,1,n);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        if (x==1) modify(1,y,z);
        else printf("%I64d\n",query(1,y,z));
    }
}