士兵杀敌（四）
时间限制：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
线段树超时 唉！！明天看看树状数组 先贴上
树状数组点我！！！
树状数组：#include<stdio.h> #include<string.h> #include<queue> #include<algorithm> #define MAX 1000010 int n,m; int sum[MAX<<1]; int lowdit(int x) { return x&(-x);//求小于x的最大的2的k次方是多少 } void update(int x,int a)//更新当前点之前的所有点 { while(x>0) { sum[x]+=a; x-=lowdit(x); } } int find(int x)//查询当前点 { int ans=0; while(x<=m) { ans+=sum[x]; x+=lowdit(x); } return ans; } int main() { int j,i; int a,b,c; char op[10]; scanf("%d%d",&n,&m); memset(sum,0,sizeof(sum)); for(i=1;i<=n;i++) { scanf("%s",op); if(op[0]=='A') { scanf("%d%d%d",&a,&b,&c); update(a-1,-c); update(b,c); } else { scanf("%d",&a); printf("%d ",find(a)); } } return 0; }
线段树超时代码：#include<stdio.h> #include<string.h> #include<queue> #include<algorithm> #define MAX 1000010 #define INF 0x3f3f3f int n,m; int sum[MAX<<2];//记录每个区间的值，即每个气球被标记的次数 int add[MAX<<2];//记录每个对应区间的每个数的值 int ans; void pushup(int o)//向上传递用来更新每一次被标记区间的值 的改变 { sum[o]=sum[o<<1]+sum[o<<1|1]; } void pushdown(int o,int m)//向下传递 { if(add[o]) { add[o<<1]+=add[o]; add[o<<1|1]+=add[o]; sum[o<<1]+=add[o]*(m-(m>>1));//线段树左分支的区间长度等于右分支或者比右分支大1 sum[o<<1|1]+=add[o]*(m>>1); add[o]=0; } } void gettree(int o,int l,int r)//建树 { sum[o]=add[o]=0; if(l==r) return ; int mid=(l+r)>>1; gettree(o<<1,l,mid); gettree(o<<1|1,mid+1,r); pushup(o); } void update(int o,int l,int r,int L,int R,int v) { if(L<=l&&R>=r) { add[o]+=v; sum[o]+=v*(r-l+1); return ; } pushdown(o,r-l+1); int mid=(l+r)>>1; if(L <= mid) update(o<<1, l, mid, L ,R,v); if(R > mid) update(o<<1|1, mid+1, r, L ,R,v); pushup(o);//维护区间的值 } int find(int o,int l,int r,int L,int R)//查找每个气球被修改次数并输出 { if(L<=l&&R>=r) { return sum[o]; } pushdown(o,r-l+1); int ans=0; int mid=(r+l)>>1; if(L<=mid) ans+=find(o<<1,l,mid,L,R); if(R>mid) ans+=find(o<<1|1,mid+1,r,L,R); return ans; } int main() { int n,m,j,i; int a,b,c; char op[10]; while(scanf("%d%d",&n,&m)!=EOF) { gettree(1,1,n); for(i=1;i<=n;i++) { scanf("%s%d",op,&a); if(op[0]=='A') { scanf("%d%d",&b,&c); update(1,1,n,a,b,c); } else printf("%lld ",find(1,1,n,a,a)); } } return 0; }