树状数组与线段树模板 JAVA

Acwing1264 动态求连续区间和

给定 nn 个数组成的一个数列,规定有两种操作,一是修改某个元素,二是求子数列 [a,b][a,b] 的连续和。

输入格式

第一行包含两个整数 nn 和 mm,分别表示数的个数和操作次数。

第二行包含 nn 个整数,表示完整数列。

接下来 mm 行,每行包含三个整数 k,a,bk,a,b (k=0k=0,表示求子数列[a,b][a,b]的和;k=1k=1,表示第 aa 个数加 bb)。

数列从 11 开始计数。

输出格式

输出若干行数字,表示 k=0k=0 时,对应的子数列 [a,b][a,b] 的连续和。

数据范围

1≤n≤1000001≤n≤100000,
1≤m≤1000001≤m≤100000,
1≤a≤b≤n1≤a≤b≤n

输入样例:

10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8

输出样例:

11
30
35

注:因数据量较大,因此在输入时选择了BufferedReader  

树状数组解法: 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.IOException;
public class Main{
	static final int N = 100005;
	static int []arr = new int [N + 5];
	static int []tree = new int [N + 5];
	static int n,m;
	public static void main(String[] args) throws IOException {
		BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
		String []s1 = reader.readLine().split(" ");
		n = Integer.parseInt(s1[0]);
		m = Integer.parseInt(s1[1]);
		String []s2 = reader.readLine().split(" ");
		for(int i = 1;i <= n;i ++) {
			arr[i] = Integer.parseInt(s2[i-1]);
		}
		for(int i = 1;i <= n;i ++) {
			add(i,arr[i]);
		}
		for(int i = 0;i < m;i ++) {
			String []s3 = reader.readLine().split(" ");
			int k = Integer.parseInt(s3[0]);
			int a = Integer.parseInt(s3[1]);
			int b = Integer.parseInt(s3[2]);
			if(k == 0) {
				System.out.println(query(b) - query(a-1));
			}else {
				add(a,b);
			}
		}
	}
	static int lowbit(int x) {
		return x & (-x);
	}
	static void add(int x,int v) {
		for(int i = x;i <= n;i += lowbit(i)) {
			tree[i] += v;
		}
	}
	static int query(int x) {
		int res = 0;
		for(int i = x;i > 0;i -= lowbit(i)) {
			res += tree[i];
		}
		return res;
	}
}

 

线段树解法

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.IOException;
public class Main{
	static final int N = 100005;
	static int []arr = new int [N + 5];
	static int []tree = new int [4*N + 5];
	public static void main(String[] args) throws IOException {
		BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
		String []s1 = reader.readLine().split(" ");
		int n = Integer.parseInt(s1[0]);
		int m = Integer.parseInt(s1[1]);
		String []s2 = reader.readLine().split(" ");
		for(int i = 1;i <= n;i ++) {
			arr[i] = Integer.parseInt(s2[i-1]);
		}
		build(1,1,n);
		for(int i = 0;i < m;i ++) {
			String []s3 = reader.readLine().split(" ");
			int k = Integer.parseInt(s3[0]);
			int a = Integer.parseInt(s3[1]);
			int b = Integer.parseInt(s3[2]);
			if(k == 0) {
				System.out.println(query(1,1,n,a,b));
			}else {
				update(1,1,n,a,b);
			}
		}
	}
	static void build(int node,int L,int R) {
		if(L == R) {
			tree[node] = arr[L];
			return;
		}
		int mid = (L + R) >> 1;
		int left_node  = node * 2 + 1;
		int right_node = node * 2 + 2;
		build(left_node,L,mid);
		build(right_node,mid+1,R);
		tree[node] = tree[left_node] + tree[right_node];
	}
	static void update(int node,int L,int R,int idx,int val) {
		if(L == R) {
			arr[idx] += val;
			tree[node] = arr[idx];
			return;
		}
		int mid = (L + R) >> 1;
		int left_node  = node * 2 + 1;
		int right_node = node * 2 + 2;
		if(idx >= L && idx <= mid) {
			update(left_node,L,mid,idx,val);
		}
		else {
			update(right_node,mid+1,R,idx,val);
		}
		tree[node] = tree[left_node] + tree[right_node];
	}
	static int query(int node,int L,int R,int x,int y) {
		if(x > R || y < L) {
			return 0;
		}
		if(x <= L && y >= R) {
			return tree[node];
		}
		int mid = (L + R) >> 1;
		int left_node  = node * 2 + 1;
		int right_node = node * 2 + 2;
		int left_sum  = query(left_node,L,mid,x,y);
		int right_sum = query(right_node,mid+1,R,x,y);
		return left_sum + right_sum;
	}
}

 

发布了125 篇原创文章 · 获赞 9 · 访问量 6960
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览