2022河南萌新联赛第(一)场:河南工业大学 G - 热身小游戏

G - 热身小游戏

线段树
q q q 次操作看成一个长度为 q q q 的序列,初始值都是 1 1 1

对于第 i i i 次操作,如果是操作 1 1 1 则将下标为 i i i 的位置修改为 a a a,对于操作 2 2 2 则进行区间修改为 1 1 1,对于操作 3 3 3 就是区间求乘积。

时间复杂度为 O ( n log ⁡ n ) O(n\log{n}) O(nlogn)

#include
using namespace std;

typedef long long LL;
const int N = 100010, MOD = 1e9 + 7;

struct Node
{
	int l,r,sum,lazy;
}tr[N*4];

void pushup(int u)
{
	tr[u].sum=(LL)tr[u<<1].sum*tr[u<<1|1].sum%MOD;
}

void pushdown(int u)
{
	auto &root=tr[u],&left=tr[u<<1],&right=tr[u<<1|1];
	if(root.lazy)
	{
		left.sum=right.sum=root.sum;
		left.lazy=right.lazy=root.lazy;
		root.lazy=0;
	}
}

void build(int u,int l,int r)
{
	if(l==r) tr[u]={l,r,1,0};
	else
	{
		tr[u]={l,r,1,0};
		int mid=l+r>>1;
		build(u<<1,l,mid),
		build(u<<1|1,mid+1,r);
		pushup(u);
	}
}

void modify(int u,int l,int r,int d)
{
	if(tr[u].l>=l&&tr[u].r<=r)
	{
		tr[u].sum=d;
		tr[u].lazy=d;
	}
	else
	{
		pushdown(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid) modify(u<<1,l,r,d);
		if(r>mid) modify(u<<1|1,l,r,d);
		pushup(u);
	}
}

int main()
{
	int q;cin>>q;
	
	build(1,1,N-10);
	
	for(int i=1;i<=q;i++)
	{
		int op;cin>>op;
		if(op==1)
		{
			int a;cin>>a;
			modify(1,i,i,a);
		}
		if(op==2)
		{
			int l,r;cin>>l>>r;
			modify(1,l,r,1);
		}
		if(op==3) cout<<tr[1].sum<<endl;
	}
	return 0;
}

你可能感兴趣的:(#,数据结构进阶,算法,数据结构,c++)