nyoj 123 士兵杀敌（四） 树状数组【单点查询+区间修改】

士兵杀敌（四）

时间限制：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;
}