hdu 1166 敌兵布阵

树状数组:

#include<stdio.h>
#include<string.h>
using namespace std;
int n,a[500010];
char sh[500];
int lowbit(int i)
{
	return i&(-i);
}
int sum(int i)
{
	int sum=0;
	while(i>0)
	{
		sum+=a[i];i-=lowbit(i);
	}
	return sum;
}
void update(int i,int val)
{
	while(i<=n)
	{
		a[i]+=val;i+=lowbit(i);
	}
}
int main()
{
	int val,m,i,j,k,x,y,t,z=1;
	scanf("%d",&t);
	while(t--)
	{
		memset(a,0,sizeof(a));
		scanf("%d",&n);
		for(i=1;i<=n;i++)
		{
			scanf("%d",&val);update(i,val);
		}
		printf("Case %d:\n",z++);
		while(scanf("%s",sh)!=EOF)
		{
			if(sh[0]=='E') break;
			scanf("%d %d",&x,&y);
			if(sh[0]=='A') update(x,y);
			else if(sh[0]=='S') update(x,-y);
			else printf("%d\n",sum(y)-sum(x-1));
		}
	}
	return 0;
}

线段树:

#include<stdio.h>
#include<string.h>
using namespace std;
struct Tree
{
	int l,r,mid;
	int  sum;
}tree[150010];
char commend[30];
int ai[50020];
void build(int i,int a,int b)
{
	tree[i].l=a;
	tree[i].r=b;
	tree[i].mid=(a+b)/2;
	tree[i].sum=0;
	if(a==b)
	tree[i].sum=ai[a];
	else
	{
		build(2*i,a,(a+b)/2);
		build(2*i+1,(a+b)/2+1,b);
		tree[i].sum=tree[2*i].sum+tree[2*i+1].sum;	
	}		
}
void update(int i,int a,int b,int c)
{
	if(a==tree[i].l&&b==tree[i].r)
	{
		tree[i].sum+=c;
		return;
	}
	if(b<=tree[i].mid)
    update(2*i,a,b,c);
    else if(tree[i].mid<a)
    update(2*i+1,a,b,c);
    else
    {
    	update(2*i,a,tree[i].mid,c);
    	update(2*i+1,tree[i].mid+1,b,c);
    }
    tree[i].sum=tree[2*i].sum+tree[2*i+1].sum;
}
int query(int i,int a,int b)
{
	if(a<=tree[i].l&&tree[i].r<=b)
	return tree[i].sum;
	if(b<=tree[i].mid)
	return query(2*i,a,b);
	else if(tree[i].mid<a)
	return query(2*i+1,a,b);
	else 
	return query(2*i,a,tree[i].mid)+query(2*i+1,tree[i].mid+1,b);
}
int main()
{
	int t,n,m,i,j,k,a,b,c,z=1;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		printf("Case %d:\n",z++);
		for(i=1;i<=n;i++)
		{
			scanf("%d",&ai[i]);
		}
		build(1,1,n);
		while(scanf("%s",commend)!=EOF)
		{
			if(commend[0]=='E') break;
			else if(commend[0]=='A')
			{
				scanf("%d%d",&a,&c);
				update(1,a,a,c);
			}
			else if(commend[0]=='S')
			{
				scanf("%d%d",&a,&c);
				update(1,a,a,-c);
			}
			else if(commend[0]=='Q')
			{
				scanf("%d%d",&a,&b);
				printf("%d\n",query(1,a,b));
			}
		}
		
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值