acm_123士兵杀敌4

来源：互联网发布：sql配置管理器在哪编辑：程序博客网时间：2024/06/05 02:22

士兵杀敌（四）

时间限制：2000 ms | 内存限制：65535KB

难度：5

描述

南将军麾下有百万精兵，现已知共有M个士兵，编号为1~M，每次有任务的时候，总会有一批编号连在一起人请战（编号相近的人经常在一块，相互之间比较熟悉），最终他们获得的军功，也将会平分到每个人身上，这样，有时候，计算他们中的哪一个人到底有多少军功就是一个比较困难的事情，军师小工的任务就是在南将军询问他某个人的军功的时候，快速的报出此人的军功，请你编写一个程序来帮助小工吧。

假设起始时所有人的军功都是0.

输入

只有一组测试数据。
每一行是两个整数T和M表示共有T条指令，M个士兵。（1<=T,M<=1000000)
随后的T行，每行是一个指令。
指令分为两种：
一种形如
ADD 100 500 55表示，第100个人到第500个人请战，最终每人平均获得了55军功，每次每人获得的军功数不会超过100，不会低于-100。
第二种形如：
QUERY 300 表示南将军在询问第300个人的军功是多少。

输出

对于每次查询输出此人的军功，每个查询的输出占一行。

样例输入

4 10ADD 1 3 10QUERY 3ADD 2 6 50QUERY 3

样例输出

来源

[张云聪]原创

上传者

张云聪

技巧是使用线段树进行延迟跟新，缩短每次跟新所需要的操作，最后的query才需从O（1）提升到log（n）的复杂度来替换掉跟新的消耗

import java.util.Scanner;

public class SoldierKillEnemy4 {

    publicstatic void main(String[] args) {
      SoldierKillEnemy4 soldierKillEnemy4 = newSoldierKillEnemy4();

      soldierKillEnemy4.solution();
    }

    int[]tree;
    intsize;

    public voidsolution() {
       Scanner in = new Scanner(System.in);
       int cmds = in.nextInt();
       size = in.nextInt();
       tree = newint[size * 4 + 1];

       Stringcommand = "" ;
       int temp =0;
       while(cmds> 0){
          command =in.next();
         if(command.endsWith("ADD")){
            Update(in.nextInt(), in.nextInt(), in.nextInt());
          }else{
             temp =in.nextInt();
            System.out.println(Query(temp, temp));
          }
         cmds--;
       }
    }

    //成段跟新
    public voidUpdate(int n, int m, int value) {
       Update(1, 1,size, n, m, value);
    }

    private voidUpdate(int index, int left, int right, int n, int m, int value){
       if (left>=n && right<= m) {
          tree[index]+= value;
         return;
       } else{
          tree[index<< 1] += tree[index];
          tree[index<< 1|1] += tree[index];
          tree[index]= 0;
          int mid =(left + right) >> 1;
          if (mid>= n && left <= m) {
             Update(index<< 1, left, mid, n, m, value);
          }
          if(mid <m && right >= n){
             Update(index<< 1 | 1, mid + 1, right, n, m, value);
          }
       }
    }

    //差选线段起止之间的权值和
    public intQuery(int start, int end) {
       returnQuery(1, 1, size, start, end);
    }

   private int Query(int index,int left, int right, int start, int end) {
       //System.out.println(left +" "+ right);
       if(tree[index] != 0) {
          //找出相交的线段长直接把乘积返回
          int length =(end > right ? right : end)
                - (left >start ? left : start) + 1;
          length =length >= 0 ? length : 0;
          returnlength * tree[index];
       } else{
          //节点自身为0
          if (left ==right) {
             returntree[index];
          }
          int sum =0;
          //搜索范围为一个线段，用二分法搜索相交范围
          int mid =(left + right) >> 1;
          if (mid>= start && left <= end) {
             sum +=Query(index << 1, left, mid, start, end);
          }
          if (mid <end && right >= start) {
             sum +=Query(index << 1 | 1, mid + 1, right, start, end);
          }
          returnsum;
       }
    }
}

0 0