http://acm.hdu.edu.cn/showproblem.php?pid=1754
生涯A的第一道线段树的题
线段树裸题,维护区间最大值,点修改和区间查询
数组写法:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#define INF 0x3f3f3f
using namespace std;
const int N=200010;
int ans,n,m,a[N];
char s[5];
struct node
{
int l,r,w;
}tree[4*N];
void build(int root,int l,int r)
{
tree[root].l=l;tree[root].r=r;
if(l==r)
{
tree[root].w=a[l];
return ;
}
int mid=(l+r)/2;
build(root*2,l,mid);
build(root*2+1,mid+1,r);
tree[root].w=max(tree[root*2].w,tree[root*2+1].w);
}
void query(int k,int l,int r)
{
if(tree[k].l>=l&&tree[k].r<=r)
{
ans=max(tree[k].w,ans);
return ;
}
int mid=(tree[k].l+tree[k].r)/2;
if(l<=mid) query(k*2,l,r);
if(r>mid) query(k*2+1,l,r);
}
void add(int k,int p,int q)
{
if(tree[k].l==tree[k].r)
{
if(tree[k].l==p)
tree[k].w=q;
return ;
}
int mid=(tree[k].l+tree[k].r)/2;
if(p<=mid) add(k*2,p,q);
else add(k*2+1,p,q);
tree[k].w=max(tree[k*2].w,tree[k*2+1].w);
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
int x,y;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,1,n);
while(m--)
{
int x,y;
scanf("%s%d%d",s,&x,&y);
if(s[0]=='Q')
{
ans=-INF;
query(1,x,y);
printf("%d\n",ans);
}
if(s[0]=='U')
{
add(1,x,y);
}
}
}
return 0;
}
指针写法:
用指针写的时候要注意多组数据,每组数据结束要记得delete不然会MLE。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=200010;
int a[N],n,m,x,y;
char s[10];
struct Seg{
int v;
Seg *ls,*rs;
}*root;
//建树
void build(Seg* &p,int l,int r)
{
p=new Seg();
p->v=0;
if(l==r)
{
p->v=a[l];
return ;
}
int mid=(l+r)/2;
build(p->ls,l,mid);
build(p->rs,mid+1,r);
p->v=max(p->ls->v,p->rs->v);
}
//单点修改a[k]=val
void modify(Seg* &p,int l,int r,int k,int val)
{
if(k<l||k>r) return ;
else if(l==r)
{
p->v=val;
return ;
}
else
{
int mid=(l+r)/2;
modify(p->ls,l,mid,k,val);
modify(p->rs,mid+1,r,k,val);
p->v=max(p->ls->v,p->rs->v);
}
}
//区间查询[L,R]
int query(Seg* &p,int l,int r,int L,int R)
{
if(l>R||r<L) return 0;
else if(L<=l&&r<=R) return p->v;
else
{
int mid=(l+r)/2;
return max(query(p->ls,l,mid,L,R),query(p->rs,mid+1,r,L,R));
}
}
void destruct(Seg* &p) {
if(p == NULL) return;
destruct(p -> ls);
destruct(p -> rs);
delete p;
p = NULL;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(root,1,n);
while(m--)
{
scanf("%s%d%d",s,&x,&y);
if(s[0]=='U')
modify(root,1,n,x,y);
if(s[0]=='Q')
printf("%d\n",query(root,1,n,x,y));
}
destruct(root);
}
return 0;
}