士兵杀敌（四）
时间限制：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\n",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\n",find(1,1,n,a,a));
}
}
return 0;
}