萌新求助线段树,样例都没过/kk
查看原帖
萌新求助线段树,样例都没过/kk
342013
ILE自动机楼主2020/4/30 11:42
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int n,m;
typedef long long ll;
ll In[maxn];
struct node
{
	int l,r;
	ll sum,mx;
}tree[maxn*4];
inline int ls(int i){return i<<1;}
inline int rs(int i){return i<<1|1;}
void pushup(int i)
{
	tree[i].sum=tree[ls(i)].sum+tree[rs(i)].sum;
	tree[i].mx=max(tree[ls(i)].mx,tree[rs(i)].mx);
}
inline void build(int i,int l,int r)
{
	tree[i].l=l;tree[i].r=r;
	if(l==r){tree[i].sum=tree[i].mx=In[i];return;}
	int mid=(l+r)>>1;
	build(ls(i),l,mid);
	build(rs(i),mid+1,r);
	pushup(i);
}
inline void kpf(int i,int l,int r)
{
	if(tree[i].l==tree[i].r){tree[i].sum=sqrt(tree[i].sum);tree[i].mx=sqrt(tree[i].mx);return;}
	if(tree[ls(i)].r>=l&&tree[ls(i)].mx>1)kpf(ls(i),l,r);
	if(tree[rs(i)].l<=r&&tree[rs(i)].mx>1)kpf(rs(i),l,r);
	pushup(i); 
}
inline ll get(int i,int l,int r)
{
	if(tree[i].l>=l&&tree[i].r<=r)return tree[i].sum;
	ll ans=0;
	if(tree[ls(i)].r>=l)ans+=get(ls(i),l,r);
	if(tree[rs(i)].l<=r)ans+=get(rs(i),l,r);
	return ans;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&In[i]);
	}
	build(1,1,n);
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		int cz;
		scanf("%d",&cz);
		if(cz==0)
		{
			int l,r;
			scanf("%d %d",&l,&r);
			if(l>r)swap(l,r);
			kpf(1,l,r);
		}
		else if(cz==1)
		{
			int l,r;
			scanf("%d %d",&l,&r);
			if(l>r)swap(l,r);
			printf("%lld\n",get(1,l,r));
		}
	}
	return 0;
 } 
2020/4/30 11:42
加载中...