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

//树状数组（948Ms）

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 1000010
using namespace std;
int q[N];
int n;
int lowbit(int x)
{
	return x&(-x);
}
void add(int i,int w)
{
	while(i<=n)
	{
		q[i]+=w;
		i+=lowbit(i);
	}
}
int sum(int i)
{
	int ans=0;
	while(i>0)
	{
		ans+=q[i];
		i-=lowbit(i);
	}
	return ans;
}
int main()
{
	int m,i,j;
	int x,y,w;
	char s[6];
	while(scanf("%d%d",&m,&n)!=EOF)
	{
		memset(q,0,sizeof(q));
		while(m--)
		{
			scanf("%s",s);
			if(strcmp(s,"ADD")==0)
			{
				scanf("%d%d%d",&x,&y,&w);
				add(x,w);
				add(y+1,-w);
			}
			else
			{
				scanf("%d",&x);
				printf("%d\n",sum(x));
			}
		}
	}
	return 0;
}

//线段树（1748Ms。。。）

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 1000010
using namespace std;
struct zz
{
	int l;
	int r;
	int m;
}q[N<<2];
void build(int gen,int l,int r)
{
	q[gen].l=l;
	q[gen].r=r;
	q[gen].m=0;
	if(l==r)
		return ;
	int mid=(l+r)/2;
	build(gen<<1,l,mid);
	build(gen<<1|1,mid+1,r);
}
void update(int gen,int l,int r,int w)
{
	if(l<=q[gen].l&&q[gen].r<=r)
	{
		q[gen].m+=w;
		return ;
	}
	int mid=(q[gen].l+q[gen].r)/2;
	if(r<=mid)
		update(gen<<1,l,r,w);
	else if(l>mid)
		update(gen<<1|1,l,r,w);
	else
	{
		update(gen<<1,l,mid,w);
		update(gen<<1|1,mid+1,r,w);
	}
}
int query(int gen,int x)
{
	if(q[gen].l==q[gen].r)
		return q[gen].m;
	int mid=(q[gen].l+q[gen].r)/2;
	int cnt=q[gen].m;
	if(x<=mid)
		cnt+=query(gen<<1,x);
	else
		cnt+=query(gen<<1|1,x);
	return cnt;
}
int main()
{
	int n,m,i,j;
	int x,y,c;
	char s[6];
	while(scanf("%d%d",&m,&n)!=EOF)
	{
		build(1,1,n);
		while(m--)
		{
			scanf("%s",s);
			if(strcmp(s,"ADD")==0)
			{
				scanf("%d%d%d",&x,&y,&c);
				update(1,x,y,c);
			}
			else
			{
				scanf("%d",&x);
				printf("%d\n",query(1,x));
			}
		}
	}
	return 0;
}
//一位大神写的这个，省时了不少。。

#include<stdio.h>
#include<string.h>
#define N 2300010
#define M 1048576
int a[N];
int cnt,p;
int x,y,w;
void update(int i,int l,int r)
{
	int mid=(l+r)/2;
	if(x<=l&&y>=r)
	{
		a[i]+=w;
		return ;
	}
	if(mid>=y)
		update(i<<1,l,mid);
	else if(mid<x)
		update(i<<1|1,mid+1,r);
	else
	{
		update(i<<1,l,mid);
		update(i<<1|1,mid+1,r);
	}
}
void query(int i,int l,int r)
{
	int mid=(l+r)/2;
	cnt+=a[i];
	if(l==r)
		return ;
	if(mid>=x)
		query(i<<1,l,mid);
	else
		query(i<<1|1,mid+1,r);
}
int main()
{
	int n,m;
	char s[6];
	while(scanf("%d%d",&m,&n)!=EOF)
	{
		memset(a,0,sizeof(a));
		while(m--)
		{
			scanf("%s",s);
			if(strcmp(s,"ADD")==0)
			{
				scanf("%d%d%d",&x,&y,&w);
				update(1,1,M);
			}
			else
			{
				scanf("%d",&x);
				cnt=0;
				query(1,1,M);
				printf("%d\n",cnt);
			}
		}
	}
	return 0;
}