之前学过一段时间的线段树,后来就没有再做了,上次比赛做到几道题居然都是线段树,而且渐渐发现线段树的适用性之广和重要性,而且之前学的不全面,于是决定重现学一遍线段树,顺便改一下之前的代码风格
一.单点更新
最典型的最基础的题目就是敌兵布阵了,题意大概是给n个数,然后3种操作:某个点的数加上x,某个点的数减去x,求[x,y]区间所有数的和。前两种操作都一样,只不过减去x相当于加上-x
先贴一段以前写的搓代码:
#include<stdio.h>
#include<string.h>
const int MAXNODE = 50005;
typedef struct NODE{
int left,right;
int num;
}Node;
int ans=0;
int num[50010];
Node node[140000];
void creattree(int i,int left,int right)
{
node[i].left=left;
node[i].right=right;
if(left==right)
{
node[i].num=num[right];
return;
}
creattree(i*2,left,(left+right)/2);
creattree(i*2+1,(left+right)/2+1,right);
node[i].num+=node[i*2].num+node[i*2+1].num;
}
void sum(int left,int right,int i)
{
if(node[i].left==left&&node[i].right==right)
{
ans+=node[i].num;
return;
}
else{
int mid=(node[i].left+node[i].right)/2;
if(mid>=right) sum(left,right,i*2);
else if(mid<left) sum(left,right,i*2+1);
else{
sum(left,mid,i*2);
sum(mid+1,right,i*2+1);
}
}
}
void add(int left,int right,int i,int k)
{
node[i].num+=k;
if(node[i].left==left&&node[i].right==right)
{
return;
}
else{
int mid=(node[i].left+node[i].right)/2;
if(mid>=right) add(left,right,i*2,k);
else if(mid<left) add(left,right,i*2+1,k);}
}
int main()
{
char s[10];
int t;
while(scanf("%d",&t)!=EOF)
{
for(int Case=1;Case<=t;Case++)
{
memset(num,0,sizeof(0));
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&num[i]);
}
creattree(1,1,n);
printf("Case %d:\n",Case);
while( scanf("%s",s))
{
int a,b;
if(s[0]=='E') break;
scanf("%d%d",&a,&b);
if(s[0]=='A') add(a,a,1,b);
else if(s[0]=='S') add(a,a,1,-b);
else if(s[0]=='Q')
{
ans=0;
sum(a,b,1);
printf("%d\n",ans);
}
}
}
}
return 0;
}
接着是最新更新的代码风格:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn = 5e5+5;
int sum[maxn*4];
int num[maxn];
void up(int rt){
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int rt,int l,int r){
if(l==r) {sum[rt]=num[l];return;}
int m=(l+r)>>1;
build(rt<<1,l,m);
build(rt<<1|1,m+1,r);
up(rt);
}
void update(int p,int add,int rt,int l,int r){
if(l==r) {
sum[rt]+=add;
return;
}
int m=(l+r)>>1;
if(p>m) update(p,add,rt<<1|1,m+1,r);
else update(p,add,rt<<1,l,m);
up(rt);
}
int query(int L,int R,int rt,int l,int r){
if(l>=L&&r<=R) return sum[rt];
int m=(l+r)>>1;
int ret = 0;
if(L<=m) ret+=query(L,R,rt<<1,l,m);
if(R>m) ret+=query(L,R,rt<<1|1,m+1,r);
return ret;
}
int main()
{
int t;
scanf("%d",&t);
for(int cas=1;cas<=t;cas++){
int n;
memset(sum,0,sizeof(sum));
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&num[i]);
}
build(1,1,n);
printf("Case %d:\n",cas);
char s[10];
while(scanf("%s",s)!=EOF&&strcmp(s,"End")){
int x,y;
scanf("%d%d",&x,&y);
if(s[0]=='Q') printf("%d\n",query(x,y,1,1,n));
else if(s[0]=='A') update(x,y,1,1,n);
else if(s[0]=='S') update(x,-y,1,1,n);
}
}
}
版权声明:本文为博主原创文章,未经博主允许不得转载。
原文:http://blog.csdn.net/sin_xf/article/details/47146359