title: 士兵杀敌( 一) 南阳理工 108
tags: [树状数组]
题目链接
描述
南将军手下有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分析:
最简单的树状数组的应用,树状数组主要用于求区间和。
如果不懂树状数组的应用的话,可以看这篇博客敌兵布阵
代码:
#include<stdio.h>
#include<iostream>
#include<stack>
#include<string.h>
#include<stdlib.h>
#include<queue>
using namespace std;
int N,M;
int a[1000009]= {0};///每个士兵的杀敌数目
int e[1000009]= {0};///树状数组
int zhuan(int n)
{
return (n&(-n));
}
void UpDate(int biao,int num)///加入当前的士兵的话,就要把含有当前士兵的区间全部更新
{
while(biao<=N)
{
e[biao]+=num;
biao+=zhuan(biao);
}
}
int SUM(int n)///求1~n这个区间的和
{
int sum=0;
while(n>0)
{
sum+=e[n];
n-=zhuan(n);
}
return sum;
}
int main()
{
scanf("%d%d",&N,&M);
for(int i=1; i<=N; i++)
{
scanf("%d",&a[i]);
UpDate(i,a[i]);///每次输入一个数后,都要把当前的树状数组更新
}
int n1,n2;
for(int i=1; i<=M; i++)
{
scanf("%d%d",&n1,&n2);
printf("%d\n",SUM(n2)-SUM(n1-1));
}
return 0;
}