敵兵佈陣(線段樹or樹狀數組)

這個題線段樹的板子,敲了兩個小時,還TLE,唉,狗生太難了,紀念一下昨天的情人節。

第一行一個整數T,表示有T組數據。 
每組數據第一行一個正整數N(N<=50000),表示敵人有N個工兵營地,接下來有N個正整數,第i個正整數ai代表第i個工兵營地裏開始時有ai個人(1<=ai<=50)。 
接下來每行有一條命令,命令有4種形式: 
(1) Add i j,i和j爲正整數,表示第i個營地增加j個人(j不超過30) 
(2)Sub i j ,i和j爲正整數,表示第i個營地減少j個人(j不超過30); 
(3)Query i j ,i和j爲正整數,i<=j,表示詢問第i到第j個營地的總人數; 
(4)End 表示結束,這條命令在每組數據最後出現; 
每組數據最多有40000條命令 

Output

對第i組數據,首先輸出“Case i:”和回車, 
對於每個Query詢問,輸出一個整數並回車,表示詢問的段中的總人數,這個數保持在int以內。 

Sample Input

1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End 

Sample Output

Case 1:
6
33
59

 

 

#include<iostream>
using namespace std;
long long d[100009];
void bulid(int p,int l,int r){
	if(l==r)
	{	//cout<<1<<endl;
		scanf("%lld",&d[p]);
		return ;
	}
	int min=(l+r)>>1;
	bulid(p<<1,l,min),bulid(p<<1|1,min+1,r);
	d[p]=d[p<<1]+d[p<<1|1];
}
int query(int l,int r,int s,int t,int p)
{
	if(l<=s&&t<=r)
		return d[p];
	int mid=(s+t)>>1,sum=0;
	if(r>mid)
		sum+=query(l,r,mid+1,t,p<<1|1);
	if(l<=mid)
		sum+=query(l,r,s,mid,p<<1);
	return sum;
}
void update(int flag,int num,int l,int r,int p)
{
	if(l==r&&r==flag)
	{
		d[p]+=num;
		return ;
	}
	int mid=(l+r)>>1;
	if(flag<=mid)
		update(flag,num,l,mid,p<<1);
	else
		update(flag,num,mid+1,r,p<<1|1);
	d[p]=d[p<<1]+d[p<<1|1];
}
int main(){
	int t ;
	scanf("%d",&t);
	int n;
	char a[10];
	int c1,c2;
	for(int i=1;i<=t;i++)
	{	scanf("%d",&n);
		bulid(1,1,n);	
		printf("Case %d:\n",i);
		while(scanf("%s",a)){
		if(a[0]=='E')
			break;
			scanf("%d %d",&c1,&c2);
		if(a[0]=='Q')
			printf("%d\n",query(c1,c2,1,n,1));
		else if(a[0]=='S')
			update(c1,-c2,1,n,1);
		else if(a[0]=='A') 
			update(c1,c2,1,n,1);
	}
	}
	return 0;
} 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章