CodeForces - 920F SUM and REPLAC（线段树）

题目链接：点击这里

题目大意：
给定一个长度为 $n$ 的序列，进行 $m$ 个操作，操作有两种：

1. 将 $[l,r]$ 的每一个元素变成其因子个数
2. 求 $[l,r]$ 的元素和

题目分析：
此题难点是操作 $1$
我们发现 $d(x)\le x$ ，而且很多都是远小于，所以可以大胆猜测修改到相等，即 $x=1,2$ 所用的次数不会很多
然后提前线筛求出 $d(x)$ ，然后用线段树暴力修改即可，同时维护区间最大值，如果最大值小于等于 $2$ 就说明不用修改了，否则就暴力修改
事实证明这个猜测是对的，那么为什么呢：
一个数 $x$ 的因子最多有 $2\sqrt{x}$ 个，即 $d(x)\le 2\sqrt{x}$ ，所以每次修改相当于对 $x$ 开了一次根号，那么显然会在很少次（5）修改后修改到 $2$ ，所以均摊起来就是 $O(n\log n)$ 的了

具体细节见代码：

//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<unordered_map>
#define ll long long
#define inf 0x3f3f3f3f
#define Inf 0x3f3f3f3f3f3f3f3f
//#define int  ll
#define endl '\n'
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
using namespace std;
int read()
{
	int res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 1e6+5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
int cnt,pri[maxn],d[maxn],num[maxn];
bool vis[maxn];
void get_d(int n)
{
	d[1] = 1;
	for(int i = 2;i <= n;i++)
	{
		if(!vis[i]) 
		{
			pri[++cnt] = i;
			d[i] = 2;num[i] = 1;
		}
		for(int j = 1;j <= cnt && i*pri[j] <= n;j++)
		{
			vis[i*pri[j]] = true;
			if(i%pri[j] == 0)
			{
				d[i*pri[j]] = d[i]/(num[i]+1)*(num[i]+2);
				num[i*pri[j]] = num[i]+1;
				break;
			}
			d[i*pri[j]] = d[i]*2;
			num[i*pri[j]] = 1;
		}
	}
}

int n,m;
struct node{
	int maxx; ll val;
}a[maxn<<2];
void pushup(int root)
{
	a[root].val = a[root<<1].val+a[root<<1|1].val;
	a[root].maxx = max(a[root<<1].maxx,a[root<<1|1].maxx);
}
void build(int root,int l,int r)
{
	if(l == r)
	{
		a[root].val = a[root].maxx = read();
		return ;
	}
	int mid = l+r>>1;
	build(root<<1,l,mid); build(root<<1|1,mid+1,r);
	pushup(root);
}
void update(int root,int l,int r,int ql,int qr)
{
	if(a[root].maxx <= 2 || l>qr || r<ql) return ;
	if(l == r)
	{
		a[root].val = a[root].maxx = d[a[root].val];
		return ;
	}
	int mid = l+r>>1;
	update(root<<1,l,mid,ql,qr);
	update(root<<1|1,mid+1,r,ql,qr);
	pushup(root);
}
ll query(int root,int l,int r,int ql,int qr)
{
	if(l>qr || r<ql) return 0;
	if(l>=ql && r<=qr) return a[root].val;
	int mid = l+r>>1;
	return query(root<<1,l,mid,ql,qr)+query(root<<1|1,mid+1,r,ql,qr);
}
int main()
{
	get_d(1e6);
	// for(int i = 1;i <= 20;i++) cout<<i<<":"<<d[i]<<endl;
	n = read(),m = read();
	build(1,1,n);
	while(m--)
	{
		int opt = read(),l = read(),r = read();
		if(opt == 1) update(1,1,n,l,r);
		else printf("%lld\n",query(1,1,n,l,r));
	}
	return 0;
}