社区讨论

线段树求调

学术版参与者 2已保存回复 4

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
4 条
当前快照
1 份
快照标识符
@ltqrl599
此快照首次捕获于
2024/03/14 13:03
2 年前
此快照最后确认于
2024/03/14 17:29
2 年前
查看原帖
刚学OI的萌新妹妹性感代码在线求调教(确信
CPP
#include<bits/stdc++.h>
#define int long long
using namespace std;
struct node
{
	int l,r,v,tag;
};
int n,m,a,b,c,d,var[100005],N=1;
node t[1600000];
int ls(int idx)
{
	return idx*2;
}
int rs(int idx)
{
	return idx*2+1;
}
int fa(int idx)
{
	return idx>>1;
}
int len(int idx)
{
	return t[idx].r-t[idx].l+1;
}
void addtag(int idx,int num)
{
	t[idx].tag=num;
	t[idx].v+=num*len(idx);
	return;
}
void merge(int idx)
{
	t[idx].v=t[ls(idx)].v+t[rs(idx)].v;
	return;
}
void push_down(int idx)
{
	addtag(ls(idx),t[idx].tag);
	addtag(rs(idx),t[idx].tag);
	t[idx].tag=0;
	return;
}
bool all_in(int l1,int r1,int l2,int r2)
{
	if(l1>=l2 and r1<=r2)return true;
	return false;
}
bool all_out(int l1,int r1,int l2,int r2)
{
	if(l1>r2 or r1<l2)return true;
	return false;
}
int query(int L,int R,int idx)
{
	if(all_in(t[idx].l,t[idx].r,L,R))return t[idx].v;
	if(all_out(t[idx].l,t[idx].r,L,R))return 0;
	push_down(idx);
	return query(L,R,ls(idx))+query(L,R,rs(idx));
}
void update(int L,int R,int k,int idx)
{
	if(all_in(t[idx].l,t[idx].r,L,R))
	{
		addtag(idx,k);
		return;
	}
	if(all_out(t[idx].l,t[idx].r,L,R))return;
	update(L,R,k,ls(idx));
	update(L,R,k,rs(idx));
	merge(idx);
	return;
}
void start()
{
	cin>>n>>m;
	while(n>N)N*=2;
	for(int i=1;i<=n;i++)cin>>var[i];
	t[1].l=1;
	t[1].r=N;
	for(int i=1;i<N;i++)
	{
		t[ls(i)].l=t[i].l;
		t[ls(i)].r=t[i].l+(len(i)>>1)-1;
		t[rs(i)].l=t[ls(i)].r+1;
		t[rs(i)].r=t[i].r;
	}
	for(int i=N;i<=2*N-1;i++)t[i].v=var[i-N+1];
	for(int i=N-1;i>=1;i--)merge(i);
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	start();
	for(int i=1;i<=m;i++)
	{
		cin>>a>>b>>c;
		if(a==1)
		{
			cin>>d;
			update(b,c,d,1);
		}
		if(a==2)cout<<query(b,c,1)<<endl;
	}
	return 0;
}

回复

4 条回复,欢迎继续交流。

正在加载回复...