树状数组与线段树模板 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;
	}
}