线段树--模板

线段树的命名tree

线段树的创建 build

线段树的查询 query

10月4日的线段树模板(现更新)
问题:不够简洁
#include<iostream>
using namespace std;
const int maxind=256;
int segTree[maxind*4+10];
int array[maxind];
bool flag=false;
inline min(int a,int b){return a<b;}
void build(int node ,int begin,int end)//
{
if(begin==end)
 segTree[node]=array[begin];
else 
{
build(2*node,begin,(begin+end)/2);//先建立左子树
build(2*node+1,(begin+end)/2+1,end);//建立右子树
if(segTree[2*node]<=segTree[2*node+1])
{
 segTree[node]=segTree[2*node];//由左子树的完成并带来信息,来进行操作,和更新
}
else segTree[node]=segTree[2*node+1];

}
}
int query(int node,int begin ,int end,int left,int right)
{

 if(left>end||right<begin)//[begin,end]和[left,right]没有交集的时候,剪枝
  return -1;
 int p1,p2;
 if(begin>=left&&end<=right)//把所有属于[left,right]集合的元素都提取有效的信息
  return segTree[node];
 p1=query(2*node,begin,(begin+end)/2,left,right);//对左子树进行递归查询
 p2=query(2*node+1,(begin+end)/2+1,end,left,right);//对右子树进行递归查询
 if(p1==-1)return p2;//当左子树没有交集的时候,我们将期望右子树
 if(p2==-1)return p1;
 if(p1<=p2) return p1;//左右子树均返回了有用的信息
 return p2;

}
void updata(int node ,int begin,int end ,int ind ,int add)//和更新的思想是一样的
{
if(begin==end&&end==ind)//找到了下标为ind的那个线段点,就特定情况而触发的效果
{segTree[node]+=add;return ;}
int m=(begin+end)>>1;
if(ind<=m) updata(2*node,begin,m,ind,add);//如果在左子树上面,就继续搜索左子树
else updata(2*node+1,m+1,end,ind,add);
segTree[node]=min(segTree[2*node],segTree[2*node+1]);//由左子树和右子树的更新来更新这个树

}
int main()
{
array[0]=1,array[1]=2,array[2]=2,array[3]=4,array[4]=1,array[5]=3;
build(1,0,5);
for(int i=1;i<=20;i++)
   cout<<"seg"<<i<<" "<<segTree[i]<<endl;

cout<<"查询功能"<<endl;
 int ans=query(1,0,5,2,2);
cout<<"result:"<<ans<<endl;
updata(1,0,5,2,-1);
cout<<"查询功能"<<endl;
 ans=query(1,0,5,2,2);
cout<<"result:"<<ans<<endl;
return 0;
2018-3-20的线段树模板(参考哈理工的模板)
#include<cstdio>
#include<stdlib.h>
const int maxn=200000;
int val[maxn]={5,9,3,4,6,6,4,8,9,3,11}; 
struct node{
	int max;
	int left,right;//val的区间范围 
}tree[maxn*3];
int max(int a,int b){return (a>b)?a:b;}
int build(int root,int begin,int end){
	tree[root].left=begin;
	tree[root].right=end;
	if(begin==end)
		return tree[root].max=val[begin];
	int mid=(begin+end)/2;
	int a=build(root*2,begin,mid);
	int b=build(root*2+1,mid+1,end);
	return tree[root].max=max(a,b);
}
int query(int root,int begin,int end){//有交集时候,必然可以分解若干区间 
	if(tree[root].right<begin||tree[root].left>end)//不想交 
		return 0;
	if(tree[root].left>=begin&&tree[root].right<=end)
		return tree[root].max;
	int a=query(2*root,begin,end);
	int b=query(2*root+1,begin,end);
	return max(a,b);
}
int main(){
	
	build(1,0,10);
	printf("%d\n",query(1,4,6));
	
	return 0;
} 


猜你喜欢

转载自blog.csdn.net/kala0/article/details/79756267