【南陽OJ108】士兵殺敵(一)(線段樹)

士兵殺敵一

士兵殺敵(一)
時間限制:1000 ms | 內存限制:65535 KB
難度:3
描述
南將軍手下有N個士兵,分別編號1到N,這些士兵的殺敵數都是已知的。

小工是南將軍手下的軍師,南將軍現在想知道第m號到第n號士兵的總殺敵數,請你幫助小工來回答南將軍吧。

注意,南將軍可能會問很多次問題。

輸入
只有一組測試數據
第一行是兩個整數N,M,其中N表示士兵的個數(1< N<1000000),M表示南將軍詢問的次數(1< M<100000)
隨後的一行是N個整數,ai表示第i號士兵殺敵數目。(0<=ai<=100)
隨後的M行每行有兩個整數m,n,表示南將軍想知道第m號到第n號士兵的總殺敵數(1<=m,n<=N)。
輸出
對於每一個詢問,輸出總殺敵數
每個輸出佔一行
樣例輸入
5 2
1 2 3 4 5
1 3
2 4
樣例輸出
6
9

線段樹比較具體的代碼說明:http://blog.csdn.net/rricky_/article/details/77126832
這題就是直接運用

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct Node
{
    int l,r,sum;
}soldier[1000000<<2];     //四倍 空間 

void pushup(int o)
{
    soldier[o].sum=soldier[o*2].sum+soldier[o*2+1].sum;
}

void build(int o,int l,int r)
{
    soldier[o].l=l;
    soldier[o].r=r;
    if(l==r)
    {
        int t;
        scanf("%d",&t);
        soldier[o].sum=t;
        return;
    }
    int mid=(l+r)/2;
    build(o*2,l,mid);
    build(o*2+1,mid+1,r);
    pushup(o);
}

int kill(int o,int l,int r,int x,int y)
{
    if(l==x&&r==y)
    return soldier[o].sum;
    int mid=(l+r)/2;
    if(mid>=y)
    return kill(o*2,l,mid,x,y);
    else if(mid<x)
    return 
        kill(o*2+1,mid+1,r,x,y);
    else return 
        kill(o*2,l,mid,x,mid)+ kill(o*2+1,mid+1,r,mid+1,y);


}

int main()
{
    int m,n,M,N;
    scanf("%d%d",&m,&n);
    build(1,1,m);
    for(int i=0;i<n;i++)
    {
        scanf("%d%d",&M,&N);
        printf("%d\n",kill(1,1,m,M,N));
    }
return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章