CJOJ 2440 大话西游

大话西游

Description

“大话西游” 是一个在中国非常流行的在线游戏， 由 NIE 公司开发和维护。 这个游戏来源于著名的小说《西游记》 和周星弛的电影， 游戏的背景故事充满奇幻色彩， 引人入胜。
游戏里面有很多片区域， 不同的区域由不同的统治者管辖， 其中有一个地方名叫“树国”，由一个妖怪控制着。 这里有 N 个城堡， 每个城堡都有其重要程度值（一个正整数， 不超过10^8）， 这些城堡被 N-1 条双向道路所连接， 任意两个城堡均可互达， 城堡的重要程度值是可变的。 现在， 妖怪想知道如果破坏其中的一条道路会发生什么。 本题中， 你总共需要处理Q 条指令， 每一个都具有下面所述的格式：

• (1)CHANGE i w 本指令的含义为： 将第 i 个城堡的重要程度值变为 w（ 1<=w<=10^8）
• (2)QUERY j 本指令的含义为： 输出 min1*max1+min2*max2 的值， 详细如下：
  第 j 条道路可以把“树国” 分成两个连通块， 分别称为 part1 和 part2， 其中
• min1 为 part1 中的最小重要程度值；
• max1 为 part1 中的最大重要程度值；
• min2 为 part2 中的最小重要程度值；
• max2 为 part2 中的最大重要程度值。

Input

第一行有两个整数 N(2<=N<=100000)和 Q(1<=Q<=100000),分别表示城堡的个数及指令的数目 。
接下来的一行有 N 个整数(正整数， 不超过 10^8)， 表示起初每一个城堡的重要程度值(城堡的编号为 1～N)。
接下来有 N-1 行， 每行有两个整数 u,v， 表示在城堡 u 和城堡 v 之间有一条无向边相连，（边的编号依次为 1~N-1）。
接下来有 Q 行， 每行有一个指令， 格式如下所述。

Output

对于每个"QUERY"指令， 在单独一行输出结果。

Sample Input

5 3
1 2 3 4 5
1 2
2 3
3 4
4 5
QUERY 1
CHANGE 1 10
QUERY 1

Sample Output

11
110

Hint

数据范围：
对于 30%的数据 N≤ 100 Q<=1000
对于 60%的数据 N≤ 10000 Q<=10000
对于 100%的数据 N≤ 100000 Q<=100000

Source

dfs序，线段树，树链剖分

Solution

树链剖分的板子题，只是在求和的时候注意一下分成三段考虑

Code

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <stack>
#include <map>
#include <vector>
#include <queue>
#define inf 2147483647
#define ls (x << 1)
#define rs (x << 1 | 1)
#define mid ((l + r) >> 1)
#define L 100010
#define LL long long
using namespace std;

struct node {
  int to, nxt;
} e[L << 1];
int n, q, u, v, tot, head[L], temp, e1[L], e2[L], cnt, siz[L], son[L], top[L], dfn[L], p[L], num;
LL w, ta[L << 2], ti[L << 2], A[L];
char ch[20];

inline void add(int u, int v) {
  e[++cnt].nxt = head[u], e[cnt].to = v, head[u] = cnt;
}

inline void dfs1(int x, int fa) {
  siz[x] = 1;
  for (int i = head[x]; i; i = e[i].nxt) {
    int y = e[i].to;
    if (siz[y]) continue;
    dfs1(y, x);
    siz[x] += siz[y];
    if (siz[y] > siz[son[x]]) son[x] = y;
  }
}

inline void dfs2(int x, int tp) {
  top[x] = tp, dfn[x] = ++tot, p[tot] = x;
  if (son[x]) dfs2(son[x], tp);
  for (int i = head[x]; i; i = e[i].nxt)
    if (!dfn[e[i].to]) dfs2(e[i].to, e[i].to);
}

inline void buildtree(int x, int l, int r) {
  if(l == r) {
    ta[x] = ti[x] = A[p[l]]; return ;
  }
  buildtree(ls, l, mid), buildtree(rs, mid + 1, r);
  ta[x] = max(ta[ls], ta[rs]), ti[x] = min(ti[ls], ti[rs]);
}
inline void update(int x, int l, int r, int u, LL v) {
  if (l == r) {
    ta[x] = ti[x] = w; return ;
  }
  if (u <= mid) update(ls, l, mid, u, v);
  else update(rs, mid + 1, r, u, v);
  ta[x] = max(ta[ls], ta[rs]), ti[x] = min(ti[ls], ti[rs]);
}

inline LL getmax(int x, int l, int r, int u, int v) {
  if (v < u) return -inf;
  if (u <= l && r <= v) return ta[x];
  if (v <= mid) return getmax(ls, l, mid, u, v);
  else if (u > mid) return getmax(rs, mid + 1, r, u, v);
  else return max(getmax(ls, l, mid, u, mid), getmax(rs, mid + 1, r, mid + 1, v));
}

inline LL getmin(int x, int l, int r, int u, int v) {
  if (v < u) return inf;
  if (u <= l && r <= v) return ti[x];
  if (v <= mid) return getmin(ls, l, mid, u, v);
  else if (u > mid) return getmin(rs, mid + 1, r, u, v);
  else return min(getmin(ls, l, mid, u, mid), getmin(rs, mid + 1, r, mid + 1, v));
}

int main() {
  freopen("westward.in", "r", stdin);
  freopen("westward.out", "w", stdout);
  scanf("%d %d", &n, &q);
  for (int i = 1; i <= n; ++i) scanf("%lld", &A[i]);
  for (int i = 1; i < n; ++i) scanf("%d %d", &e1[i], &e2[i]), add(e1[i], e2[i]), add(e2[i], e1[i]);
  dfs1(1, 1), dfs2(1, 1), buildtree(1, 1, n);
  for (int i = 1; i <= q; ++i) {
    cin >> ch;
    if (ch[0] == 'C') {
      scanf("%d %lld", &temp, &w);
      update(1, 1, n, dfn[temp], w);
    }
    else {
      scanf("%d", &temp);
      u = e1[temp], v = e2[temp];
      if (dfn[u] < dfn[v]) swap(u, v);
      LL max1 = getmax(1, 1, n, dfn[u], dfn[u] + siz[u] - 1);
      LL min1 = getmin(1, 1, n, dfn[u], dfn[u] + siz[u] - 1);
      LL max2 = max(getmax(1, 1, n, 1, dfn[u] - 1), getmax(1, 1, n, dfn[u] + siz[u], n));
      LL min2 = min(getmin(1, 1, n, 1, dfn[u] - 1), getmin(1, 1, n, dfn[u] + siz[u], n));
      LL ans = max1 * min1 + max2 * min2;
      printf("%lld\n", ans);
    }
  }
  return 0;
}

Summary

打的时候一直RE，后来才发现线段树的数组开小了，然后就是变量名看错了