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 longusing 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，后来才发现线段树的数组开小了，然后就是变量名看错了