描述
南将军手下有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 21 2 3 4 51 32 4 样例输出
- 6
- 9
- //
- 解法一:非树状AC方式:
-
#include<stdio.h> int a[1000000]; int main() { int N,M; int i,n,m; scanf("%d %d",&N,&M); a[0]=0; for(i=1;i<N+1;i++) { scanf("%d",&a[i]); a[i]+=a[i-1]; } for(i=0;i<M;i++) { scanf("%d %d",&n,&m); printf("%d\n",a[m]-a[n-1]); } return 0; }
-
解法二:树状数组AC方式:
#include<stdio.h> #include<string.h> int a[1000000]; int main() { int n,sum; scanf("%d %d",&n,&sum); int i,j,k; memset(a,0,sizeof(a)); for(i=1;i<=n;i++) { int num; scanf("%d",&num); j=i; while(j<=n) { a[j]=a[j]+num; j+=j&(-j); } } for(i=0;i<sum;i++) { scanf("%d %d",&k,&j); int s1=0,s2=0; k=k-1; while(k>=1) { s1=s1+a[k]; k-=k&(-k); } while(j>=1) { s2=s2+a[j]; j-=j&(-j);//或者j-=j&(j^(j-1)); } printf("%d\n",s2-s1); } return 0; }