nyoj123 士兵杀敌四（区间更新 & 单点访问）

 

士兵杀敌（四）

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

难度： 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 10
ADD 1 3 10
QUERY 3
ADD 2 6 50
QUERY 3   样例输出 10
60

算法分析：起初我是链表建的树超时了好多次，后来改用数组也是险过，开数组大小时也得是原来的 3 倍；

 

View Code

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<cstring>
#define N 1000010

using namespace std;

struct SegTree
{
int left;
int right;
int amount;
    SegTree()
    {
        amount=0;
    }
}tree[4*N];
int ans;

void BuildSegTree(int root,int L,int R)
{
    tree[root].left=L;
    tree[root].right=R;
    tree[root].amount=0;
if(L<R)
    {
int M=(L+R)/2;
        BuildSegTree(2*root,L,M);
        BuildSegTree(2*root+1,M+1,R);
    }
}

void AddSegTree(int root,int L,int R,int C)
{
if(L<=tree[root].left&&tree[root].right<=R)
    {
        tree[root].amount+=C;
return ;
    }
int M=(tree[root].left+tree[root].right)/2;
if(R<=M)  AddSegTree(2*root,L,R,C);
if(L>M)   AddSegTree(2*root+1,L,R,C);
if(L<=M&&M<R)  
    {
        AddSegTree(2*root,L,M,C);
        AddSegTree(2*root+1,M+1,R,C);
    }
}

void find_e(int root,int i)
{
if(tree[root].left<=i&&i<=tree[root].right)
        ans+=tree[root].amount;
if(tree[root].left==tree[root].right)  return ;
int M=(tree[root].left+tree[root].right)/2;
if(i<=M)  find_e(2*root,i);
if(i>M)   find_e(2*root+1,i);
}

int main()
{
int n,m,i;
    scanf("%d%d",&m,&n);
    BuildSegTree(1,1,n);
char s[10];
for(i=1;i<=m;i++)
    {
        scanf("%s",s);
if(s[0]=='A') 
        {
int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            AddSegTree(1,a,b,c);
        }
else 
        {
int i;
            scanf("%d",&i);
            ans=0;
            find_e(1,i);
            printf("%d\n",ans);
        }
    }
return 0;
}

 

转载于:https://www.cnblogs.com/zhourongqing/archive/2011/10/19/2217974.html