杭电oj 1166 敌兵补阵 线段树java模板

import java.util.ArrayList;
import java.util.Scanner;

class treeNode{
	int left;
	int right;
	int data;
}
public class h1166 {
	
	// 求左节点和右节点的和
	/**
	 * 
	 * @param idxroot
	 * @param left 左边界
	 * @param right 右边界
	 * @param segTree 线段树数组
	 * @param arr  输入的原始数据
	 */
	public static void build(int idxroot,int left,int right,treeNode segTree[],int arr[]) {
		 segTree[idxroot].left=left;
		 segTree[idxroot].right=right;
		if(left==right) {
			segTree[idxroot].data=arr[left];
			return;
		}
		int mid=(left+right)/2;
		build(idxroot*2,left,mid,segTree,arr);//构建左子树
		build(idxroot*2+1,mid+1,right,segTree,arr);//构建右子树
		//更新当前节点
		segTree[idxroot].data=segTree[idxroot*2].data+segTree[idxroot*2+1].data;
	}
	
	/**
	 * 
	 * @param idxroot 根节点
	 * @param left  左边界
	 * @param right 右边界
	 * @param data  修改的数据
	 * @param segTree  线段树数组
	 * @param index 叶子节点的边界的索引,例如（[0,0],[1,1],[2,2]...[n-1,n-1]）:
	 */
	public static void update(int idxroot,int left,int right,int data,treeNode segTree[],int index) {
		if(left==right&&left==index) {
			segTree[idxroot].data+=data;		
			return;
		}
		int mid=(left+right)/2;
		//小在左边，大在右边
		if(index<=mid) {
			//递归左边
			update(idxroot*2,left,mid,data,segTree,index);
		}else {
			//递归右边
			update(idxroot*2+1,mid+1,right,data,segTree,index);
		}
		segTree[idxroot].data=segTree[idxroot*2].data+segTree[idxroot*2+1].data;
	}
	/**
	 * 
	 * @param segTree 线段树数组
	 * @param idxroot 根节点
	 * @param ql 查询的左边界
	 * @param qr 查询的右边界
	 * @param left 线段树的左边界
	 * @param right 线段树的右边界
	 * @return
	 */
	public static long  query(treeNode segTree[],int idxroot,int ql,int qr,int left,int right) {
		if(left>=ql&&right<=qr) {
			return segTree[idxroot].data;
		}
		int mid=(left+right)/2;
		long sum=0;
		if(mid>=ql) {
			//左子树有重合
			sum+=query(segTree,idxroot*2,ql,qr,left,mid);
		}
		if(mid<qr) {
			sum+=query(segTree,idxroot*2+1,ql,qr,mid+1,right);
		}
		return sum;
	}
	public static void main(String[] args) {
		// TODO Auto-generated method stub
             Scanner sc=new Scanner(System.in);
             int t=sc.nextInt();
             sc.nextLine();
             int c=1;
             while(t-->0) {
            	 int n=sc.nextInt();
            	 int[] arr=new int[n];
            	 for(int i=0;i<arr.length;i++) {
            		 arr[i]=sc.nextInt();
            	 }
            	 sc.nextLine();
            	
            	 treeNode tree[]=new treeNode[n*4];
            	 for(int i=0;i<tree.length;i++) {
            		 tree[i]=new treeNode();
            	 }
            	//构建线段树
            	 build(1,0,n-1,tree,arr);
            	 ArrayList<Long> l=new ArrayList<>();
            	 while(sc.hasNext()) {
            		 String ar[]=sc.nextLine().split(" ");
            		 if(ar.length==1&&ar[0].equals("End")) {
            			 break;
            		 }
            		 if(ar[0].equals("Query")) {
            			 int ql=Integer.parseInt(ar[1])-1;
            			 int qr=Integer.parseInt(ar[2])-1;
            			l.add(query(tree,1,ql,qr,0,n-1));
            		 }
            		 if(ar[0].equals("Add")) {
            			 int index=Integer.parseInt(ar[1])-1;
            			 int data=Integer.parseInt(ar[2]);
            			 update(1,0,n-1,data,tree,index);
            		 }
            		 if(ar[0].equals("Sub")) {
            			 int index=Integer.parseInt(ar[1])-1;
            			 int data=Integer.parseInt(ar[2])*-1;
            			 update(1,0,n-1,data,tree,index);
            		
            		 }
            	 }
        
            	 System.out.println("Case "+c+":");
            	 for(int i=0;i<l.size();i++) {
            		 System.out.println(l.get(i));
            	 }
            	 l.clear();
            	 c++;
             }
	}

}