【線段樹】小白逛公園

题目描述

　　小新经常陪小白去公园玩，也就是所谓的遛狗啦…在小新家附近有一条“公园路”，路的一边从南到北依次排着n个公园，小白早就看花了眼，自己也不清楚该去哪些公园玩了。
　　一开始，小白就根据公园的风景给每个公园打了分-.-。小新为了省事，每次遛狗的时候都会事先规定一个范围，小白只可以选择第a个和第b个公园之间（包括a、b两个公园）选择连续的一些公园玩。小白当然希望选出的公园的分数总和尽量高咯。同时，由于一些公园的景观会有所改变，所以，小白的打分也可能会有一些变化。
　　那么，就请你来帮小白选择公园吧。
输入格式

　　第一行，两个整数N和M，分别表示表示公园的数量和操作（遛狗或者改变打分）总数。
　　接下来N行，每行一个整数，依次给出小白 开始时对公园的打分。
　　接下来M行，每行三个整数。第一个整数K，1或2。K=1表示，小新要带小白出去玩，接下来的两个整数a和b给出了选择公园的范围（1≤a,b≤N）；K=2表示，小白改变了对某个公园的打分，接下来的两个整数p和s，表示小白对第p个公园的打分变成了s（1≤p≤N）。
　　其中，1≤N≤500 000，1≤M≤100 000，所有打分都是绝对值不超过1000的整数。
输出格式

　　小白每出去玩一次，都对应输出一行，只包含一个整数，表示小白可以选出的公园得分和的最大值。
样例输入
5 3
1 2 -3 4 5
1 2 3
2 2 -1
1 2 3
样例输出
2
-1 

一道線段樹的入門題。
用四個標記維護每個節點的狀態：
tree[p].maxL表示該區間內包含最左端元素的最大和；
tree[p].maxR表示該區間內包含最右端元素的最大和；
tree[p].sum表示该区间的綜合；
tree[p].maxS表示該區間的最大子區間和。

那麼每個p節點的狀態與其左子樹lc和右子樹rc的關係為：
tree[p].maxL = max(lc.maxL, lc.sum + rc.maxL)；
tree[p].maxR = max(rc.maxR, rc.sum + lc.maxR)；
tree[p].sum = lc.sum + rc.sum；
tree[p].maxS = max(lc.maxS, rc.maxS,
    lc.maxR + rc.maxL)。

具體細節建程序。

Accode：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <bitset>

using std::max;

const char fi[] = "rqnoj626.in";
const char fo[] = "rqnoj626.out";
const int maxN = 500010;
const int maxM = 100010;
const int MAX = 0x3fffff00;
const int MIN = -MAX;

struct SegTree
{
  int L, R, lc, rc;
  int maxL, maxR, sum, maxS;
};

SegTree tree[maxN << 2];
int sc[maxN];
int n, m, tot;

  void init_file()
  {
    freopen(fi, "r", stdin);
    freopen(fo, "w", stdout);
  }
  
  void update(SegTree *ths, SegTree *lc, SegTree *rc)
  {
    ths -> maxL = max(lc -> maxL,
      lc -> sum + rc -> maxL);
    ths -> maxR = max(rc -> maxR,
      rc -> sum + lc -> maxR);
    ths -> sum = lc -> sum + rc -> sum;
    ths -> maxS = max(lc -> maxR + rc -> maxL,
      max(lc -> maxS, rc -> maxS));
  } //根據左子樹lc和右子樹rc更新ths節點的狀態。
  
  void Build(int L, int R)
  {
    int Now = ++tot;
    tree[Now].L = L;
    tree[Now].R = R;
    if (L == R)
    {
      tree[Now].maxL = tree[Now].maxR
        = tree[Now].sum = tree[Now].maxS
        = sc[L];
      return;
    }
	//若到葉節點則直接賦初始打分並返回。
    int Mid = (L + R) >> 1;
    if (L < R)
    {
      tree[Now].lc = tot + 1;
      Build(L, Mid);
	//先建左子樹。
      tree[Now].rc = tot + 1;
      Build(Mid + 1, R);
	//後建右子樹。
    }
    update(tree + Now, tree + tree[Now].lc,
      tree + tree[Now].rc);
	//根據左右子樹的信息更新當前節點。
  }
  
  void Modify(int Now, int p)
  {
    if (tree[Now].L == tree[Now].R
      && tree[Now].L == p)
    {
      tree[Now].maxL = tree[Now].maxR
        = tree[Now].sum = tree[Now].maxS
        = sc[p];
      return;
    }
	//邊界，若找到p則直接修改。
    int Mid = (tree[Now].L + tree[Now].R) >> 1;
    if (p <= Mid) Modify(tree[Now].lc, p);
	//若p在左區間則修改左子樹。
    if (p > Mid) Modify(tree[Now].rc, p);
	//若p在右區間則修改右子樹。
    update(tree + Now, tree + tree[Now].lc,
      tree + tree[Now].rc);
	//更新當前節點。
  }
  
  SegTree Find(int Now, int L, int R)
  {
    if (L <= tree[Now].L && R >= tree[Now].R)
      return tree[Now];
	//若被完全覆蓋則直接返回樹上當前節點的狀態。
    int Mid = (tree[Now].L + tree[Now].R) >> 1;
    if (R <= Mid)
      return Find(tree[Now].lc, L, R);
	//若完全被左子樹所在區間覆蓋，
	//則直接返回左子樹的狀態。
    if (L > Mid)
      return Find(tree[Now].rc, L, R);
	//同理單獨處理右子樹。
    SegTree ans;
    SegTree lc = Find(tree[Now].lc, L, R);
	//找到左子樹的狀態。
    SegTree rc = Find(tree[Now].rc, L, R);
	//找到右子樹的狀態。
    update(&ans, &lc, &rc);
	//更新當前節點的狀態。
    return ans;
  } //返回在Now節點，區間[L, R]中找到最優狀態。
  
  void work()
  {
    scanf("%d%d", &n, &m);
    for (int i = 1; i < n + 1; ++i)
      scanf("%d", sc + i);
    tot = 0;
    Build(1, n);
	//建樹並初始化各節點的狀態（根據最初打分）。
    for (; m; --m)
    {
      int O, x, y;
      scanf("%d%d%d", &O, &x, &y);
      switch (O)
      {
        case 1 :
          if (x > y) std::swap(x, y);
	//注意有可能x < y。
          printf("%d\n", Find(1, x, y).maxS);         
          break;
        case 2 :
          if (sc[x] == y) break;
          sc[x] = y;
          Modify(1, x);
          break;
      }
    }
  }
  
int main()
{
  init_file();
  work();
  exit(0);
}