大话西游
“大话西游” 是一个在中国非常流行的在线游戏, 由 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 中的最大重要程度值。
第一行有两个整数 N(2<=N<=100000)和 Q(1<=Q<=100000),分别表示城堡的个数及指令的数目 。 接下来的一行有 N 个整数(正整数, 不超过 10^8), 表示起初每一个城堡的重要程度值(城堡的编号为 1~N)。 接下来有 N-1 行, 每行有两个整数 u,v, 表示在城堡 u 和城堡 v 之间有一条无向边相连,(边的编号依次为 1~N-1)。 接下来有 Q 行, 每行有一个指令, 格式如下所述。
对于每个"QUERY"指令, 在单独一行输出结果。
5 3 1 2 3 4 5 1 2 2 3 3 4 4 5 QUERY 1 CHANGE 1 10 QUERY 1
11 110
数据范围: 对于 30%的数据 N≤ 100 Q<=1000 对于 60%的数据 N≤ 10000 Q<=10000 对于 100%的数据 N≤ 100000 Q<=100000
dfs序,线段树,树链剖分
树链剖分的板子题,只是在求和的时候注意一下分成三段考虑
#include #include #include #include #include #include #include #include #include #include #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; }
打的时候一直RE,后来才发现线段树的数组开小了,然后就是变量名看错了