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
Case 1: 6 33 59
#include <iostream>
#include <cstdio>
#include <cstring>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;
const double eps = 1e-6;
const double pi = acos(-1.0);
const int INF = 0x3f3f3f3f;
const int MOD = 1000000007;
#define ll long long
#define CL(a,b) memset(a,b,sizeof(a))
#define MAXN 100010
struct node
{
int a,b,sum;
}t[400010];
int ans, r[500010];
void build(int x, int y, int num)
{
t[num].a = x;
t[num].b = y;
if(x == y) t[num].sum = r[y];
else
{
int mid = (x+y)/2;
build(x, mid, num*2);
build(mid+1, y, num*2+1);
t[num].sum = t[num*2].sum + t[num*2+1].sum;
}
}
void add(int x, int y, int num)
{
if(t[num].a == x && t[num].b == x)
{
t[num].sum += y;
return ;
}
int mid = (t[num].a+t[num].b)/2;
if(x > mid)
add(x, y, num*2+1);
else
add(x, y, num*2);
t[num].sum = t[num*2].sum + t[num*2+1].sum;
}
void sub(int x, int y, int num)
{
if(t[num].a == x && t[num].b == x)
{
t[num].sum -= y;
return ;
}
int mid = (t[num].a+t[num].b)/2;
if(x > mid)
sub(x, y, num*2+1);
else
sub(x, y, num*2);
t[num].sum = t[num*2].sum + t[num*2+1].sum;
}
void query(int x, int y, int num)
{
if(t[num].a == x && t[num].b == y)
{
ans += t[num].sum;
return ;
}
int mid = (t[num].a+t[num].b)/2;
if(y <= mid) query(x, y, num*2);
else if(x >= mid+1) query(x, y, num*2+1);
else
{
query(x, mid, num*2);
query(mid+1, y, num*2+1);
}
}
int main()
{
int T,n;
char str[6];
scanf("%d",&T);
for(int cas=1; cas<=T; cas++)
{
int x,y;
scanf("%d",&n);
r[0] = 0;
for(int i=1; i<=n; i++)
scanf("%d",&r[i]);
build(1, n, 1);
printf("Case %d:\n",cas);
while(scanf("%s",str)==1)
{
//scanf("%d%d",&x,&y);
if(strcmp(str, "End")==0) break;
else if(strcmp(str, "Add")==0)
{
scanf("%d%d",&x,&y);
add(x, y, 1);
}
else if(strcmp(str, "Sub")==0)
{
scanf("%d%d",&x,&y);
sub(x, y, 1);
}
else if(strcmp(str, "Query")==0)
{
scanf("%d%d",&x,&y);
ans = 0;
query(x, y, 1);
printf("%d\n",ans);
}
}
}
return 0;
}
原文:http://blog.csdn.net/d_x_d/article/details/50261143