链接:click here~~
题意:
第一行一个整数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
【解题思路】单点更新:最最基础的线段树,只更新叶子节点,然后把信息用PushUP(int r)这个函数更新上来
代码:
#include#include#include#include#include#include#include#include#include#includeusing namespace std;int dir[4][2]= {{1,0},{-1,0},{0,1},{0,-1}};#define mem(a) memset(a,0,sizeof(a))#define LL long long#define Max(a,b) a>b?a:b;#define Min(a,b) a>1;
build(lchild);
build(rchild);
Pushup(rt);
}
void update(int p,int add,int l,int r,int rt) // 单点更新
{
if(l==r)
{
tree[rt]+=add;
return ;
}
int m=(l+r)>>1;
if(p<=m) update(p,add,lchild);
else update(p,add,rchild);
Pushup(rt);
}
int query(int L,int R,int l,int r,int rt) //区间查询
{
if(L<=l&&r<=R)
{
return tree[rt];
}
int m=(l+r)>>1;
int sum=0;
if(L<=m) sum+=query(L,R,lchild);
if(R>m) sum+=query(L,R,rchild);
return sum;
}
int scan()
{
int res = 0, flag = 0;
char ch;
if((ch = getchar()) == '-') flag = 1;
else if(ch >= '0' && ch <= '9') res = ch - '0';
while((ch = getchar()) >= '0' && ch <= '9')
res = res * 10 + (ch - '0');
return flag ? -res : res;
}
int main()
{
int t,n,m,i,j,tot=1;
scanf("%d",&t);
while(t--)
{
printf("Case %d:\n",tot++);
n=scan();
build(1,n,1);
char op[10];
while(scanf("%s",op))
{
if(op[0]=='E') break;
int a,b;
a=scan();
b=scan();
if(op[0]=='Q') printf("%d\n",query(a,b,1,n,1));
else if(op[0]=='S') update(a,-b,1,n,1);
else update(a,b,1,n,1);
}
}
return 0;
}
网友评论