NYOJ 123 士兵杀敌（四） （线段树）

题目链接

描述

南将军麾下有百万精兵，现已知共有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 10
  ADD 1 3 10
  QUERY 3
  ADD 2 6 50
  QUERY 3
• 样例输出
  10
  60

分析：

刚开始想到肯定要用线段树来写，然后用普通的线段树写，每次更新的时候从当前区间开始更新到最大区间，但是这样的话时间超，然后就像了一种优化点的方法

每次在更新的时候，只有找到这个特定的区间之后，才将该区间的value值加，不用向上加。然后在求值的时候每个包含该点的区间的value值都应该加上，因为他可能是通过任意一个区间去建立军功值的。

代码：

    #include<stdio.h>
    #include<iostream>
    #include<stack>
    #include<string.h>
    #include<stdlib.h>
    #include<queue>
    #include<algorithm>
    using namespace std;
    int e[10000]= {0};
    int T,M;
    struct Node
    {
        int left;
        int right;
        int value;
    } tree[4000000+10];
    
    void built(int root,int n1,int n2)///刚开始建树的时候，每个节点的军功值都是0
    {
        tree[root].left=n1;
        tree[root].right=n2;
        tree[root].value=0;
        if(n1==n2)
            return ;
        int mid=(n1+n2)/2;
        built(root*2,n1,mid);
        built(root*2+1,mid+1,n2);
    }
    void UpDate(int root,int n1,int n2,int num)
    {
        if(tree[root].left==n1&&tree[root].right==n2)///找到特定的区间后才把该区间中的军功值加上，注意这里不用累计往上加
        {
            tree[root].value+=num;///也就是以为着这个区间上的每个人的军功值都加了value
            return ;
        }
        if(n1>=tree[root*2+1].left)
        {
            UpDate(root*2+1,n1,n2,num);
        }
        else if(n2<=tree[root*2].right)
        {
            UpDate(root*2,n1,n2,num);
        }
        else
        {
            int mid=(tree[root].left+tree[root].right)/2;
            UpDate(root*2,n1,mid,num);
            UpDate(root*2+1,mid+1,n2,num);
        }
    }
    
    int Sum(int root,int n1,int sum)
    {
    
        sum+=tree[root].value;///因为在插入的时候只在特定的区间上加，所以只要找到包含这个点的区间，也就意味这该区间中有这个人的军功值
        if(tree[root].left==n1&&tree[root].right==n1)
        {
            return sum;
        }
        else if(n1>=tree[root*2+1].left)
            return (Sum(root*2+1,n1,sum));
        else if(n1<=tree[root*2].right)
            return (Sum(root*2,n1,sum));
    }
    int main()
    {
        scanf("%d%d",&T,&M);
        char a[10];  
        int n1,n2,num;
        built(1,1,M);
        for(int i=0; i<T; i++)
        {
            scanf(" %s",a);
            if(a[0]=='A')
            {
                scanf("%d%d%d",&n1,&n2,&num);
                UpDate(1,n1,n2,num);
            }
            else
            {
                scanf("%d",&n1);
                printf("%d\n",Sum(1,n1,0));
            }
        }
        return 0;
    }