java版线段树（HDU1166）

//注意事项：一定要用BufferedReader！！！！用同样代码Scanner直接超时，中间的实现增加的类直接从模板改的，也可以实现区间增加

敌兵布阵

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 97998    Accepted Submission(s): 41471

Problem Description

C国的死对头A国这段时间正在进行军事演习，所以C国间谍头子Derek和他手下Tidy又开始忙乎了。A国在海岸线沿直线布置了N个工兵营地,Derek和Tidy的任务就是要监视这些工兵营地的活动情况。由于采取了某种先进的监测手段，所以每个工兵营地的人数C国都掌握的一清二楚,每个工兵营地的人数都有可能发生变动，可能增加或减少若干人手,但这些都逃不过C国的监视。
中央情报局要研究敌人究竟演习什么战术,所以Tidy要随时向Derek汇报某一段连续的工兵营地一共有多少人,例如Derek问:“Tidy,马上汇报第3个营地到第10个营地共有多少人!”Tidy就要马上开始计算这一段的总人数并汇报。但敌兵营地的人数经常变动，而Derek每次询问的段都不一样，所以Tidy不得不每次都一个一个营地的去数，很快就精疲力尽了，Derek对Tidy的计算速度越来越不满:"你个死肥仔，算得这么慢，我炒你鱿鱼!”Tidy想：“你自己来算算看，这可真是一项累人的工作!我恨不得你炒我鱿鱼呢!”无奈之下，Tidy只好打电话向计算机专家Windbreaker求救,Windbreaker说：“死肥仔，叫你平时做多点acm题和看多点算法书，现在尝到苦果了吧!”Tidy说："我知错了。。。"但Windbreaker已经挂掉电话了。Tidy很苦恼，这么算他真的会崩溃的，聪明的读者，你能写个程序帮他完成这项工作吗？不过如果你的程序效率不够高的话，Tidy还是会受到Derek的责骂的.

 

Input

第一行一个整数T，表示有T组数据。
每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。
接下来每行有一条命令，命令有4种形式：
(1) Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30）
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人（j不超过30）;
(3)Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;
(4)End 表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令

 

Output

对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

 

Sample Input

1101 2 3 4 5 6 7 8 9 10Query 1 3Add 3 6Query 2 7Sub 10 2Add 6 3Query 3 10End 

 

Sample Output

Case 1:63359

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
    public static final int N = 100005;
    public static int tot;
    public static int[] a ;
    public static int[] tleft ;
    public static int[] tright ;
    public static long[] tsum;
    public static long[] td;

    public static void main(String[] args) throws NumberFormatException,
            IOException {

        BufferedReader sc = new BufferedReader(new InputStreamReader(System.in));
        int T = Integer.parseInt(sc.readLine());
        u: for (int k = 1; k <= T; k++) {
            tot = 0;
            a= new int[N];
            tright= new int[3 * N];
            tleft= new int[3 * N];
            tsum= new long[3 * N];
            td= new long[3 * N];
            int n, left, right, value;
            String str;
            n = Integer.parseInt(sc.readLine());
            String temp = sc.readLine();
            String shu[] = temp.split(" ");
            for (int i = 1; i <= shu.length; i++) {
                a[i] = Integer.parseInt(shu[i - 1]);
            }
            System.out.println("Case " + k + ":");
            creatTree(1, 1, n);
            for (int i = 0; i < 1000000; i++) {
                String s = sc.readLine();
                String arr[] = s.split(" ");
                str = arr[0];
                if (str.equals("Query")) {// 求区间值的和
                    left = Integer.parseInt(arr[1]);
                    right = Integer.parseInt(arr[2]);

                    System.out.println(query(1, left, right, 0));
                } else if (str.equals("Add")) {// 将区间内的数加上个值（如需给某个值增加让left=right）
                    left = Integer.parseInt(arr[1]);
                    right = left;
                    value = Integer.parseInt(arr[2]);
                    insert(1, left, right, value);
                } else if (str.equals("Sub")) {
                    left = Integer.parseInt(arr[1]);
                    right = left;
                    value = Integer.parseInt(arr[2]);

                    insert(1, left, right, value * (-1));

                } else {
                    continue u;
                }
            }
        }
    }

    public static long creatTree(int now, int left, int right) {
        // 建树类，a[]存放数据
        if (now > tot) {
            tot = now;
        }
        tleft[now] = left;
        tright[now] = right;
        long lSum = 0, rSum = 0;
        if (left < right) {
            lSum = creatTree(2 * now, left, (left + right) / 2);
            rSum = creatTree(2 * now + 1, (left + right) / 2 + 1, right);
            tsum[now] = lSum + rSum;
        } else
            tsum[now] = a[left];
        return tsum[now];
    }

    public static void insert(int now, int left, int right, int value) {
        // 将left到right区间内的数加上value，
        if (now > tot)
            return;
        if (left <= tleft[now] && right >= tright[now]) {
            td[now] += value;
            return;
        }
        long lSum = 0, rSum = 0;
        if (left <= (tleft[now] + tright[now]) / 2)
            insert(2 * now, left, right, value);
        if (right > (tleft[now] + tright[now]) / 2)
            insert(2 * now + 1, left, right, value);
        if (2 * now <= tot)
            lSum = tsum[2 * now] + td[2 * now]
                    * (tright[2 * now] - tleft[2 * now] + 1);
        if (2 * now + 1 <= tot)
            rSum = tsum[2 * now + 1] + td[2 * now + 1]
                    * (tright[2 * now + 1] - tleft[2 * now + 1] + 1);
        tsum[now] = lSum + rSum;
    }

    public static long query(int now, int left, int right, long d) {
        // 求区间left到right的值得和
        if (now > tot)
            return 0;
        if (left <= tleft[now] && right >= tright[now])
            return tsum[now] + (td[now] + d) * (tright[now] - tleft[now] + 1);
        long lSum = 0, rSum = 0;
        if (left <= (tleft[now] + tright[now]) / 2)
            lSum = query(2 * now, left, right, d + td[now]);
        if (right > (tleft[now] + tright[now]) / 2)
            rSum = query(2 * now + 1, left, right, d + td[now]);
        return lSum + rSum;
    }
}