HDU - 1166

敌兵布阵

树状数组和线段树均可，这些板子敲出来就是爽！！！
链接：HDU - 1166

树状数组做法

---

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
int t,n;
const int N = 5e4+10;
int arr[N];
int lowbit(int x)
{
	return x&-x;
}
void add(int idx,int x)
{
	for(int i=idx;i<=n;i+=lowbit(i))
	{
		arr[i]+=x;
	}
}
int sum(int idx)
{
	int res=0;
	for(int i=idx;i;i-=lowbit(i))
	{
		res+=arr[i];
	}
	return res;
}
int main()
{
	scanf("%d",&t);
	int tex=1;
	while(t--)
	{
	scanf("%d",&n);
	memset(arr,0,sizeof(arr));
	for(int i=1;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		add(i,x);
	}
	string s="";
	vector<int>ans;
//	ans.clear();
	while((cin>>s)&&s!="End")
	{
		if(s=="Query")
		{
			int l,r;
			scanf("%d %d",&l,&r);
			ans.push_back(sum(r)-sum(l-1));
		}
		else if(s=="Add")
		{
			int id,x;
			scanf("%d %d",&id,&x);
			add(id,x);
		}
		else if(s=="Sub")
		{
			int id,x;
			scanf("%d %d",&id,&x);
			add(id,-x);
		}
			}
			printf("Case %d:\n",tex++);
			for(int i=0;i<ans.size();i++)
			{
				printf("%d\n",ans[i]);
					}		
	}
	
	return 0;
}

---

线段树做法

---

#include<bits/stdc++.h>
using namespace std;
const int N = 5e4+10;
struct node{
	int l,r;
	int sum;
}tr[N<<2];
int arr[N];
int n,t;

void pushup(int u)
{
	tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}

void build(int u,int l,int r)
{
	if(l==r)
	{
		tr[u]={l,r,arr[l]};
	}
	else
	{
		tr[u]={l,r,0};
		int mid=l+r>>1;
		build(u<<1,l,mid);
		build(u<<1|1,mid+1,r);
		pushup(u);
	}
}

void modify(int u,int idx,int x)
{
	if(tr[u].l==idx&&tr[u].r==idx)
	{
		tr[u].sum+=x;
	}
	else
	{
		int mid=tr[u].l+tr[u].r>>1;
		if(idx<=mid) modify(u<<1,idx,x);
		if(idx>mid) modify(u<<1|1,idx,x);
		pushup(u);
	}
}

int query(int u,int l,int r)
{
	if(tr[u].l>=l&&tr[u].r<=r)
	{
		return tr[u].sum;
	}
		int ans=0;
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid) ans+=query(u<<1,l,r);
		if(r>mid) ans+=query(u<<1|1,l,r);
		return ans;
}

int main()
{
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&arr[i]);
		}
		build(1,1,n);
	
	string s="";
	vector<int>res;
	while((cin>>s)&&s!="End")
	{
		if(s=="Query")
		{
			int l,r;
			scanf("%d %d",&l,&r);
			res.push_back(query(1,l,r));
		}
		else if(s=="Add")
		{
			int id,x;
			scanf("%d %d",&id,&x);
			modify(1,id,x);
		}
		else if(s=="Sub")
		{
			int id,x;
			scanf("%d %d",&id,&x);
			modify(1,id,-x);
		}
	}
	printf("Case %d:\n",i);
	for(int i=0;i<res.size();i++)
	{
		printf("%d\n",res[i]);
	}
	}
	return 0;
}

---