数据结构——线段树(单点更新、)

来源:互联网 发布:深圳人工智能企业 编辑:程序博客网 时间:2024/05/18 02:38

如果有错误,欢迎大神指出!!


线段树(模板):

线段树是一种二叉搜索树,与区间树相似,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。
  对于线段树中的每一个非叶子节点[a,b],它的左儿子表示的区间为[a,(a+b)/2],右儿子表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树,最后的子节点数目为N,即整个线段区间的长度。
  使用线段树可以快速的查找某一个节点在若干条线段中出现的次数,时间复杂度为O(logN)。而未优化的空间复杂度为2N,因此有时需要离散化让空间压缩。
  线段树有很多模板,而且基本上每道题都是稍稍改动或者根本不需改动就可以直接使用线段树的模板,这里提供几个相对简洁的模板:

(找了一些模板,来自于哈工大acm书)

//单点替换、单点增减、区间求和、区间最值#include <cstdio>#include <algorithm>using namespace std;#define lson l , m , rt << 1#define rson m + 1 , r , rt << 1 | 1const int maxn = 222222;int MAX[maxn<<2];int MIN[maxn<<2];int SUM[maxn<<2];int max(int a,int b){if(a>b)return a;else return b;}int min(int a,int b){if(a<b)return a;else return b;}void PushUP(int rt){  MAX[rt] = max(MAX[rt<<1] , MAX[rt<<1|1]);  MIN[rt] = min(MIN[rt<<1] , MIN[rt<<1|1]);  SUM[rt] = SUM[rt<<1] + SUM[rt<<1|1];}void build(int l,int r,int rt) {  if (l == r)    {    scanf("%d",&MAX[rt]);    MIN[rt] = MAX[rt];    SUM[rt] = MAX[rt];    //printf("mi = %d\n",MIN[rt]);  //    printf("ma = %d\n",MAX[rt]);    return ;  }  int m = (l + r) >> 1;  build(lson);  build(rson);  PushUP(rt);}void update(int p,int tihuan,int l,int r,int rt){  if (l == r) {    MAX[rt] = tihuan;    MIN[rt] = tihuan;    SUM[rt] = tihuan;    return ;  }  int m = (l + r) >> 1;  if (p <= m) update(p , tihuan ,lson);  else update(p , tihuan , rson);  PushUP(rt);}void update1(int p,int add,int l,int r,int rt){  if (l == r) {    SUM[rt] = SUM[rt] + add;    return ;  }  int m = (l + r) >> 1;  if (p <= m) update1(p , add ,lson);  else update1(p , add , rson);  PushUP(rt);}int query(int L,int R,int l,int r,int rt){  if (L <= l && r <= R)  {    return MAX[rt];  }  int m = (l + r) >> 1;  int ret = -1;  if (L <= m) ret = max(ret , query(L , R , lson));  if (R > m)  ret =  max(ret , query(L , R , rson));  return ret;}int query1(int L,int R,int l,int r,int rt){  if (L <= l && r <= R)  {    return MIN[rt];  }  int m = (l + r) >> 1;  int ret = 99999;  if (L <= m) ret = min(ret , query1(L , R , lson));  if (R > m)  ret =  min(ret , query1(L , R , rson));  return ret;}int queryhe(int L,int R,int l,int r,int rt){  if (L <= l && r <= R)  {    return SUM[rt];  }  int m = (l + r) >> 1;  int ret = 0;  if (L <= m) ret += queryhe(L , R , lson);  if (R > m)  ret +=  queryhe(L , R , rson);  return ret;}int main(){  int n , m;  while (~scanf("%d%d",&n,&m))  {    build(1 , n , 1);    while (m --) {      char op[2];      int a , b;      scanf("%s%d%d",op,&a,&b);      if (op[0] == 'Q') //区间求最大      {         /* for(int i = 1;i<=10;i++)          printf("%d ",MAX[i]);        puts("");*/        printf("%d\n",query(a , b , 1 , n , 1));      }      else if(op[0]=='U') //单点替换        update(a , b , 1 , n , 1);      else if(op[0]=='M')//区间求最小      {        /*for(int i = 1;i<=10;i++)          printf("%d ",MIN[i]);        puts("");*/        printf("%d\n",query1(a , b , 1 , n , 1));      }      else if(op[0]=='H')//区间求和      {        printf("%d\n",queryhe(a , b , 1 , n , 1));      }      else if(op[0]=='S')//单点增加      {        scanf("%d%d",&a,&b);        update1(a , b , 1 , n , 1);      }       else if(op[0]=='E')//单点减少      {        scanf("%d%d",&a,&b);        update1(a , -b , 1 , n , 1);      }    }  }  return 0;}

  • maxn是题目给的最大区间,而节点数要开4倍,确切的来说节点数要开大于maxn的最小2x的两倍
  • lson和rson分辨表示结点的左儿子和右儿子,由于每次传参数的时候都固定是这几个变量,所以可以用预定于比较方便的表示
  • 以前的写法是另外开两个个数组记录每个结点所表示的区间,其实这个区间不必保存,一边算一边传下去就行,只需要写函数的时候多两个参数,结合lson和rson的预定义可以很方便
  • PushUP(int rt)是把当前结点的信息更新到父结点
  • PushDown(int rt)是把当前结点的信息更新给儿子结点
  • rt表示当前子树的根(root),也就是当前所在的结点
建树是一个关键(如图):

对于一个父节点 i 来说,那么它的子节点2*i(左),2*i+1(右)。




0 0
原创粉丝点击