线段树模板

点的修改:

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<vector>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#define endl '\n'
#define sc(x) scanf("%d",&x)
#define rson(x) x<<1|1
#define lson(x) x<<1
using namespace std;
typedef long long LL;
const int size=2e5+5;
int arr[size];
struct node
{
	int l,r,no;
	LL maxn;
}tree[size<<2];
void build(int k,int l,int r)
{
	tree[k].l=l,tree[k].r=r;
	if(l==r)
	{
		tree[k].maxn=arr[l];
		return ;
	}
	int mid=(l+r)>>1;
	build(lson(k),l,mid);
	build(rson(k),mid+1,r);
	tree[k].maxn=max(tree[lson(k)].maxn,tree[rson(k)].maxn);
}
int  query(int k,int l,int r)
{
	int ans=0;
	if(l==tree[k].l&&r==tree[k].r)
	{
		return tree[k].maxn;
	}
	int mid=(tree[k].l+tree[k].r)>>1;
	if(l>=mid+1)
	{
		ans=query(rson(k),l,r);
	}
	else if(r<=mid)
	{
		ans=query(lson(k),l,r);
	}
	else
	{
		ans=max(query(lson(k),l,mid),query(rson(k),mid+1,r));
	}
	return ans;
}
void change(int k,int d,int x)
{
	if(tree[k].l==d&&tree[k].r==d)
	{
		tree[k].maxn=x;
		return ;
	}
	int mid=(tree[k].l+tree[k].r)>>1;
	if(d>=tree[k].l&&d<=mid) change(lson(k),d,x);
	else change(rson(k),d,x);
	tree[k].maxn=max(tree[lson(k)].maxn,tree[rson(k)].maxn);
}
int main()
{
	int n,m;
	while(~scanf("%d%d",&n,&m))
	{
		memset(tree,0,sizeof(tree));
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&arr[i]);
		}
		build(1,1,n);
		char op[2];int a,b;
		while(m--)
		{
			scanf("%s%d%d",op,&a,&b);
			if(op[0]=='Q') printf("%d\n",query(1,a,b));
			else change(1,a,b);
		}
	}
	return 0;
}

区间的修改,查询

const int size=1e5+5;
typedef long long LL;
struct node{
	int l,r;
	LL sum;
	LL tag;
}tree[size<<2];
void build(int k,int l,int r)
{
	tree[k].l=l,tree[k].r=r;
	if(l==r)
	{
		tree[k].sum=1;
		return ;
	}
	int mid(md(l,r));
	build(lson(k),l,mid);
	build(rson(k),mid+1,r);
	tree[k].sum=tree[lson(k)].sum+tree[rson(k)].sum;
}
void change(int k)
{
	if(tree[k].l!=tree[k].r)
	{
		int ls=lson(k),rs=rson(k);
		tree[ls].sum=(tree[ls].r-tree[ls].l+1)*tree[k].tag;
		tree[rs].sum=(tree[rs].r-tree[rs].l+1)*tree[k].tag;
		tree[ls].tag=tree[k].tag;
		tree[rs].tag=tree[k].tag;
	}
	tree[k].tag=0;
}
void add(int k,int l,int r,LL x)
{
	
	if(tree[k].l==l&&tree[k].r==r)
	{
		tree[k].sum=(tree[k].r-tree[k].l+1)*x;
		tree[k].tag=x;
		return ;
	}
	if(tree[k].tag) change(k);
	int mid=md(tree[k].l,tree[k].r);
	if(l>=mid+1) add(rson(k),l,r,x);
	else if(r<=mid) add(lson(k),l,r,x);
	else add(lson(k),l,mid,x),add(rson(k),mid+1,r,x);
	tree[k].sum=tree[lson(k)].sum+tree[rson(k)].sum;
}
posted @ 2018-08-15 12:34  Fly_White  阅读(109)  评论(0编辑  收藏  举报