ACM算法入门 线段树

线段树

---

---

前言

难啊……

---

一、什么是线段树

线段树，是一种二叉搜索树。它将一段区间划分为若干单位区间，每一个节点都储存着一个区间。它功能强大，支持区间求和，区间最大值，区间修改，单点修改等操作。
线段树的思想和分治思想很相像。
线段树的每一个节点都储存着一段区间[L…R]的信息，其中叶子节点L=R。它的大致思想是：将一段大区间平均地划分成2个小区间，每一个小区间都再平均分成2个更小区间……以此类推，直到每一个区间的L等于R（这样这个区间仅包含一个节点的信息，无法被划分）。通过对这些区间进行修改、查询，来实现对大区间的修改、查询。
这样一来，每一次修改、查询的时间复杂度都只为O ( log2 n )
但是，可以用线段树维护的问题必须满足区间加法，否则是不可能将大问题划分成子问题来解决的。

二、代码

代码如下（示例）：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <set>
#include <algorithm>
using namespace std;
const int N = 1e6+10;
int n, C, a[N], ans1, ans2;
struct seg {
  int l, r, mx;
}node[N<<2];
void pushup(int i) {
  node[i].mx = max(node[i<<1].mx, node[i<<1|1].mx);
}
void build(int l, int r, int i) {
  node[i] = {l, r, C};
  if (l == r) return;
  int mid = l + r >> 1;
  build(l, mid, i << 1);
  build(mid + 1, r, i << 1 | 1);
}
void query(int v, int i) {
  if (node[i].l == node[i].r) {
    if (node[i].mx == C) ans1++;
    node[i].mx -= v;
    return;
  }
  if (node[i<<1].mx >= v) query(v, i << 1);
  else query(v, i << 1 | 1);
  pushup(i);
}
int main() {
  int T;
  scanf("%d",&T);
  while (T--) {
    ans1 = ans2 = 0;
    scanf("%d%d",&n,&C);
    for (int i = 1; i <= n; i++)
      scanf("%d",&a[i]);
    build(1, n, 1);
    for (int i = 1; i <= n; i++)
      query(a[i], 1);
    multiset<int> st;
    for (int i = 1; i <= n; i++) {
      if (st.lower_bound(a[i]) != st.end()) {
        int v = *st.lower_bound(a[i]);
        st.erase(st.find(v));
        st.insert(v - a[i]);
      } else {
        st.insert(C - a[i]);
        ans2++;
      }
    }
    printf("%d %d\n",ans1, ans2);
  }
  return 0;
}

代码如下（示例）：

#include <bits/stdc++.h>
using namespace std;
#define N 1000000
int T,a[N],n,x,y,num=1;
char s[10];
struct seg
{
    int l,r,sum;
}node[N<<2];

void pushup(int i)
{
    node[i].sum=node[i<<1].sum+node[i<<1|1].sum;
}
void build(int l,int r,int i)
{
    node[i]={l,r};
    if (l==r)
    {
        node[i].sum=a[l];
        return;
    }
    int mid=l+r>>1;
    build(l,mid,i<<1);
    build(mid+1,r,i<<1|1);
    pushup(i);
}
void modify(int p,int w,int i)
{
    if (node[i].l==p&&node[i].r==p)
    {
        node[i].sum+=w;
        return;
    }
    int mid=node[i].l+node[i].r>>1;
    if(p<=mid) modify(p,w,i<<1);
    else modify(p,w,i<<1|1);
    pushup(i);
}
int query(int l,int r,int i)
{
    if (l<=node[i].l&&node[i].r<=r)
    {
        return node[i].sum;
    }
    int mid=node[i].l+node[i].r>>1;
    int res=0;
    if (l<=mid) res+=query(l,r,i<<1);
    if(r>mid) res+=query(l,r,i<<1|1);
    return res;
}
int main()
{
    cin>>T;
    while (T--)
    {
        cin>>n;
        for (int i = 1; i <= n; i++)
        {
            scanf("%d",&a[i]);
        }
        build(1,n,1);
        printf("Case %d:\n",num++);
        while (scanf("%s",s),s[0]!='E')
        {
            scanf("%d%d",&x,&y);
            if (s[0]=='A') modify(x,y,1);
            else if (s[0]=='S') modify(x,-y,1);
            else printf("%d\n",query(x,y,1));
        }       
    }
    return 0;
}

---