#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
#define LL(x) (x)<<1
#define RR(x) ((x)<<1|1)
struct Seg_tree
{
int left, right;
int num;
}t[150000];
int num[50100];
int Build(int l, int r, int root)
{
t[root].left = l;
t[root].right = r;
if(l==r)
{
return t[root].num = num[l];
}
int mid = (l+r)>>1;
return t[root].num = Build(l,mid,LL(root)) + Build(mid+1,r,RR(root));
}
void update(int id,int x,int root)
{
t[root].num += x;
if(t[root].left == t[root].right)
{
return;
}
int mid = (t[root].left+t[root].right)>>1;
if(id<=mid)
update(id,x,LL(x));
else
update(id,x,RR(x));
}
int quare(int l, int r, int root)
{
if(t[root].left==l && t[root].right==r)
return t[root].num;
int mid = (t[root].left+t[root].right)>>1;
if(r <= mid)
return quare(l,r,LL(root));
else if(mid <l)
return quare(l, r, RR(root));
else
return quare(l, mid, LL(root)) + quare(mid+1, r, RR(root));
}
int main()
{
int ncase, T;
scanf("%d",&ncase);
while(ncase--)
{
scanf("%d",&T);
for(int i=1; i<=T; i++)
scanf("%d",&num[i]);
Build(1,T,1);
printf("Case %d:\n",ncase+1);
char s[10];
while(scanf("%s",s))
{
if(strcmp(s,"End")==0) break;
int a, b;
scanf("%d %d",&a,&b);
switch(s[0])
{
case 'Q': printf("%d\n",quare(a,b,1)); break;
case 'A': update(a,b,1); break;
case 'S': update(a,-b,1); break;
}
}
}
system("pause");
return 0;
}
题目的意思就是查找区间里的结点内容,包括了线段树的查找,增加等操作,表示这样处理可以达到O(log n)的时间复杂度。
#include<cstdio>
#include<cstdlib>
#include<string>
#define LL(x) ((x)<<1)
#define RR(x) ((x)<<1|1)
#define FF(i,n) for(int i = 0 ; i < n ; i ++)
struct Seg_Tree{
int left,right,num;
int calmid() {
return (left+right)/2;
}
}tt[150000];
int num[50001];
int build(int left,int right,int idx) {
tt[idx].left = left;
tt[idx].right = right;
if(left == right) {
return tt[idx].num = num[left];
}
int mid = (left + right)/2;
return tt[idx].num = build(left,mid,LL(idx)) + build(mid+1,right,RR(idx));
}
void update(int id,int x,int idx) {
tt[idx].num += x;
if(tt[idx].left == tt[idx].right) {
return ;
}
int mid = tt[idx].calmid();
if(id <= mid) {
update(id,x,LL(idx));
} else {
update(id,x,RR(idx));
}
}
int query(int left,int right,int idx) {
if(left == tt[idx].left && right == tt[idx].right) {
return tt[idx].num;
}
int mid = tt[idx].calmid();
if(right <= mid) {
return query(left,right,LL(idx));
} else if(mid < left) {
return query(left,right,RR(idx));
} else {
return query(left,mid,LL(idx)) + query(mid+1,right,RR(idx));
}
}
int main() {
int T;
scanf("%d",&T);
FF(cas,T) {
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++) {
scanf("%d",&num[i]);
}
build(1,n,1);
printf("Case %d:\n",cas+1);
char com[9];
while(scanf("%s",com)) {
if(strcmp(com,"End") == 0) break;
int a,b;
scanf("%d%d",&a,&b);
switch(com[0]) {
case 'Q':
printf("%d\n",query(a,b,1));
break;
case 'A':
update(a,b,1);
break;
case 'S':
update(a,-b,1);
break;
}
}
}
return 0;
}
下面的还没调出来,好像是内存错误。。