士兵杀敌(一)
时间限制:
1000 ms | 内存限制:
65535 KB
难度:
3
-
描述
-
南将军手下有N个士兵,分别编号1到N,这些士兵的杀敌数都是已知的。
小工是南将军手下的军师,南将军现在想知道第m号到第n号士兵的总杀敌数,请你帮助小工来回答南将军吧。
注意,南将军可能会问很多次问题。
其实这道题不用线段树,用前缀和就能做出来,但毕竟才学线段树,就用这个题目练练手。。
代码:
前缀和
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAX=1e6+10;
int a[MAX],sum[MAX];
int main()
{
int n , m;
scanf("%d %d",&n ,&m);
for(int i = 1 ; i <= n; i++)
{
scanf("%d",&a[i]);
sum[i] = a[i] + sum[i-1];
}
int l,r;
for(int i = 0; i < m ; i++)
{
scanf("%d %d",&l , &r);
printf("%d\n",sum[r] - sum[l-1]);
}
return 0;
}
线段树:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAX=1e6+10;
struct Node
{
int l;
int r;
int sum;
}tree[MAX*4];
int Push(int o)
{
tree[o].sum = tree[o*2].sum + tree[o*2+1].sum;
}
void BuildTree(int o , int l ,int r) //建树
{
tree[o].l = l;
tree[o].r = r;
if( l == r)
{
int t;
scanf("%d",&t);
tree[o].sum = t;
return;
}
int mid = (l + r) / 2;
BuildTree(o*2 , l ,mid); //往左建树
BuildTree(o*2+1 , mid+1 , r);
Push(o);
}
int QuerySum(int o ,int l ,int r ,int x,int y)
{
if(l == x && r == y)
{
return tree[o].sum;
}
int mid = (l + r) / 2;
if(y <= mid) // 全在左边
return QuerySum(o*2 , l ,mid ,x , y);
if( x > mid)//全在右边
return QuerySum(o*2+1 , mid+1 ,r , x, y);
else
return QuerySum(o*2 , l ,mid ,x ,mid) + QuerySum(o*2+1 , mid+1 , r , mid+1 , y);
}
int main()
{
int n, m;
scanf("%d %d",&n,&m);
BuildTree(1,1,n);
int x,y;
for(int i = 0 ; i < m; i++)
{
scanf("%d %d",&x,&y);
printf("%d\n",QuerySum( 1 , 1 , n , x ,y ));
}
}