Bzoj2164 采矿（线段树+树链剖分）

题面

Bzoj

题解

对于每个节点，我们可以用树链剖分和线段树维护以下信息：

• 单独在某个点分配$i$个人的最大收益（可以$O(m)$计算）
• 分配$i$的最大收益（可以$O(m^2)$计算）

#include <cstdio>
#include <cstring>
#include <algorithm>
using std::min; using std::max;
using std::sort; using std::swap;
typedef long long ll;

template<typename T>
void read(T &x) {
    int flag = 1; x = 0; char ch = getchar();
    while(ch < '0' || ch > '9') { if(ch == '-') flag = -flag; ch = getchar(); }
    while(ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar(); x *= flag;
}

const int N = 20010, M = 51, T = 65600;
int n, m, q, X = 1 << 16, Y = ~0U >> 1, A, B, Q, op, x, y;
int fa[N], son[N], siz[N], top[N], dfn[N], dep[N], val[N], tim;
int cnt, from[N], to[N], nxt[N];
inline void addEdge(int u, int v) {
	to[++cnt] = v, nxt[cnt] = from[u], from[u] = cnt;
}
struct P {
	ll v[M];
	P() { for(int i = 0; i < M; ++i) v[i] = 0; }
	P operator + (P b) {
		P c;
		for(int i = 0; i < M; ++i) c.v[i] = max(v[i], b.v[i]);
		return c;
	}
	P operator * (P b) {
		P c;
		for(int i = 0; i < M; ++i)
			for(int j = 0; j < M - i; ++j)
				c.v[i + j] = max(c.v[i + j], v[i] + b.v[j]);
		return c;
	}
}tmp, a[N], v0[T], v1[T], s0, s1;

//读入数据
inline int getint() {
	A = ((A ^ B) + B / X + B * X) & Y;
	B = ((A ^ B) + A / X + A * X) & Y;
	return (A ^ B) % Q;
}
inline void gettmp() {
	for(int i = 1; i <= m; ++i) tmp.v[i] = getint();
	sort(&tmp.v[1], &tmp.v[m + 1]);
}

//线段树
inline void pushup(int o, int lc, int rc) { v0[o] = v0[lc] + v0[rc], v1[o] = v1[lc] * v1[rc]; }
void build(int o = 1, int l = 1, int r = n) {
	if(l == r) { v0[o] = v1[o] = a[val[l]]; return ; }
	int mid = (l + r) >> 1, lc = o << 1, rc = lc | 1;
	build(lc, l, mid), build(rc, mid + 1, r), pushup(o, lc, rc);
}
void modify(int k, int o = 1, int l = 1, int r = n) {
	if(l == r) { v0[o] = v1[o] = tmp; return ; }
	int mid = (l + r) >> 1, lc = o << 1, rc = lc | 1;
	if(k <= mid) modify(k, lc, l, mid);
	else modify(k, rc, mid + 1, r);
	pushup(o, lc, rc);
}
void query0(int ql, int qr, int o = 1, int l = 1, int r = n) {
	if(l >= ql && r <= qr) { s0 = s0 + v0[o]; return ; }
	int mid = (l + r) >> 1, lc = o << 1, rc = lc | 1;
	if(ql <= mid) query0(ql, qr, lc, l, mid);
	if(qr > mid) query0(ql, qr, rc, mid + 1, r);
}
void query1(int ql, int qr, int o = 1, int l = 1, int r = n) {
	if(l >= ql && r <= qr) { s1 = s1 * v1[o]; return ; }
	int mid = (l + r) >> 1, lc = o << 1, rc = lc | 1;
	if(ql <= mid) query1(ql, qr, lc, l, mid);
	if(qr > mid) query1(ql, qr, rc, mid + 1, r);
}

//树链剖分
void dfs(int u) {
	siz[u] = 1, dep[u] = dep[fa[u]] + 1;
	for(int i = from[u]; i; i = nxt[i]) {
		int v = to[i]; dfs(v), siz[u] += siz[v];
		if(siz[v] > siz[son[u]]) son[u] = v;
	}
}
void dfs(int u, int t) {
	dfn[u] = ++tim, val[tim] = u, top[u] = t;
	if(!son[u]) return ; dfs(son[u], t);
	for(int i = from[u]; i; i = nxt[i])
		if(to[i] != son[u]) dfs(to[i], to[i]);
}
inline void Path(int x, int y) { //dep[y] 始终小于等于 dep[x]
	if(x == y) return ;
	x = fa[x]; int fx = top[x];
	while(fx != top[y]) query0(dfn[fx], dfn[x]), x = fa[fx], fx = top[x];
	query0(dfn[y], dfn[x]);
}

int main () {
#ifdef OFFLINE_JUDGE
    freopen("233.in", "r", stdin);
    freopen("233.out", "w", stdout);
#endif
	read(n), read(m), read(A), read(B), read(Q);
	for(int i = 2; i <= n; ++i)
		read(fa[i]), addEdge(fa[i], i);
	for(int i = 1; i <= n; ++i) gettmp(), a[i] = tmp;
	dfs(1), dfs(1, 1), build(), read(q);
	while(q--) {
		read(op), read(x);
		if(!op) gettmp(), modify(dfn[x]);
		else {
			read(y);
			s0 = s1 = P();
			Path(x, y), query1(dfn[x], dfn[x] + siz[x] - 1);
			s0 = s0 * s1;
			printf("%lld\n", s0.v[m]);
		}
	}
    return 0;
}