线段树(未完成简版)

线段树

单点修改区间查询

核心

在线段树单点修改，区间查询的过程中，除了push_up外，其余三个操作都是以二分为主要思想，通过mid=l+r>>1；进而自顶向下进行建树，修改，查询。

push_up操作

static void push_up(int p) {
	tree[p]=Math.max(tree[p*2], tree[p*2+1]);
}

build建树

	static void build(int p,int l,int r) {
		if(l==r) {
			tree[p]=num[l];
			return;
		}
		int mid=l+r>>1;
		build(p*2, l, mid);
		build(p*2+1, mid+1, r);
		push_up(p);
	}

update更新单点值

static void update(int p,int l,int r,int L,int R,int date) {//p根节点,l线段树起点，r线段树终点，L查询左区间，R查询右区间，date更新的值，可以将L,R设置为相同的数字达到更新单点的作用
	if(L<=l&&R>=r) {
		tree[p]=date;
		return;
	}
	int mid=l+r>>1;
	if(mid>=L)
		update(p*2, l, mid, L, R, date);
	if(mid<R)
		update(p*2+1, mid+1, r, L, R, date);
	push_up(p);
}

query查询操作

static int query(int p,int l,int r,int L,int R) {
	int res=-N;
	if(L<=l&&R>=r) {
		return tree[p];
	}
	int mid=l+r>>1;
	if(mid>=L)
		res=Math.max(res, query(p*2, l, mid, L, R));
	if(mid<R)
		res=Math.max(res, query(p*2+1, mid+1, r, L, R));
	
	return res;
	
}

区间修改区间查询

核心

除了建树时不用push_down和addtag操作其余query和update都需要使用这两个操作，其中addtag放在递归出口条件里，push_down在每层区间内都要进行。

push_up

	static void push_up(int p) {
		tree[p]=tree[p*2]+tree[p*2+1];
	}

addtag

static void addtag(int p,int l,int r,int date) {
	tag[p]+=date;
	tree[p]+=date*(r-l+1);
}

push_down

static void push_down(int p,int l,int r) {
	if(tag[p]!=0) {
		int mid=l+r>>1;
		addtag(p*2, l, mid, tag[p]);
		addtag(p*2+1, mid+1, r, tag[p]);
		tag[p]=0;
	}
}

update

static void update(int p,int l,int r,int L,int R,int date) {
	if(L<=l&&R>=r) {
		addtag(p,l,r,date);
		return;
	}
	push_down(p,l,r);
	int mid=l+r>>1;
	
	if(L<=mid) {
		update (p*2, l, mid, L, R, date);
	}

	if(R>mid) {
		update (p*2+1, mid+1, r, L, R, date);
		}
	push_up(p);
	
}

query

static int query(int p,int l,int r,int L,int R) {
	if(L<=l&&R>=r) {
		return tree[p];
	}
	push_down(p, l, r);
	int res=0;
	int mid=l+r>>1;
	if(L<=mid)res+=query(p*2, l, mid, L, R);
	if(mid<R)res+=query(p*2+1, mid+1, r, L, R);
	return res;
}

完整代码

package 线段树;

import java.util.Scanner;

public class 区间修改_区间查询 {
	static int N=100010;
	static long num[]=new long[N];
	static int n,q;
	static long tag[]=new long[4*N];
	static long tree[]=new long[4*N];
	static void push_up(int p) {
		tree[p]=tree[p*2]+tree[p*2+1];
	}
	static void build(int p,int l,int r) {
		if(l==r) {
			tree[p]=num[l];
			return;
		}
		int mid=l+r>>1;
		build(p*2, l, mid);
		build(p*2+1, mid+1, r);
		push_up(p);
	}
	static void addtag(int p,int l,int r,long date) {
		tag[p]+=date;
		tree[p]+=date*(r-l+1);
	}
	static void push_down(int p,int l,int r) {
		if(tag[p]!=0) {
			int mid=l+r>>1;
			addtag(p*2, l, mid, tag[p]);
			addtag(p*2+1, mid+1, r, tag[p]);
			tag[p]=0;
		}
	}
	static void update(int p,int l,int r,int L,int R,long date) {
		if(L<=l&&R>=r) {
			addtag(p,l,r,date);
			return;
		}
		push_down(p,l,r);
		int mid=l+r>>1;
		
		if(L<=mid) {
			update (p*2, l, mid, L, R, date);
		}

		if(R>mid) {
			update (p*2+1, mid+1, r, L, R, date);
			}
		push_up(p);
		
	}
	
	static long query(int p,int l,int r,int L,int R) {
		if(L<=l&&R>=r) {
			return tree[p];
		}
		push_down(p, l, r);
		int res=0;
		int mid=l+r>>1;
		if(L<=mid)res+=query(p*2, l, mid, L, R);
		if(mid<R)res+=query(p*2+1, mid+1, r, L, R);
		return res;
	}
	
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		n=sc.nextInt();q=sc.nextInt();
		for(int i=1;i<=n;i++)num[i]=sc.nextLong();
		build(1, 1, n);
		
		while(q-->0) {
			char op;
			op=sc.next().charAt(0);
			if(op=='C') {
				int l,r;
				long k;
				l=sc.nextInt();r=sc.nextInt();k=sc.nextLong();
				update(1, 1, n, l, r, k);
			}else {
				int l,r;
				l=sc.nextInt();r=sc.nextInt();
				System.out.println(query(1, 1, n, l, r));
			}
		}
	}
}