bzoj4399 魔法少女LJJ 线段树合并+线段树二分+并查集

题目传送门

https://lydsy.com/JudgeOnline/problem.php?id=4399

题解

毒瘤题 \(9\) 种操作还有支持动态图的连通性

仔细读题 $ c<=7$。

哈只要加边那么动态图就变成了维护集合了。


考虑每一个集合怎们维护。

对于操作 \(1\),直接加入就可以了。

对于操作 \(2\),合并两个集合的时候直接线段树合并。注意判断两个集合是否已经联通。

对于操作 \(3\),我们取出小于 \(x\) 的数的个数,然后把小于 \(x\) 的数全部清空,把个数加到 \(x\) 上去。

对于操作 \(4\),与 \(3\) 同理。

对于操作 \(5\),线段树二分。

对于操作 \(6\),如果直接维护积的话肯定会炸 long long。所以我们可以维护它们取对数以后的值,求和维护即可。比较的时候直接比较对数和就可以了。对数底数的话任选。

对于操作 \(7\),可以直接维护出来。

对于操作 \(8, 9\),动态图


时间复杂度 \(O(m\log max_{val})\)

注意:此题卡空间!!调整好变量类型和数组大小!!!

#include<bits/stdc++.h>

#define fec(i, x, y) (int i = head[x], y = g[i].to; i; i = g[i].ne, y = g[i].to)
#define dbg(...) fprintf(stderr, __VA_ARGS__)
#define File(x) freopen(#x".in", "r", stdin), freopen(#x".out", "w", stdout)
#define fi first
#define se second
#define pb push_back

template<typename A, typename B> inline char smax(A &a, const B &b) {return a < b ? a = b, 1 : 0;}
template<typename A, typename B> inline char smin(A &a, const B &b) {return b < a ? a = b, 1 : 0;}

typedef long long ll; typedef unsigned long long ull; typedef std::pair<int, int> pii;

template<typename I> inline void read(I &x) {
	int f = 0, c;
	while (!isdigit(c = getchar())) c == '-' ? f = 1 : 0;
	x = c & 15;
	while (isdigit(c = getchar())) x = (x << 1) + (x << 3) + (c & 15);
	f ? x = -x : 0;
}

const int N = 400000 + 7;
const int INF = 1000000000;

int n, m, nod;
int rt[N], fa[N];

struct Node {
	int val, lc, rc;
	bool tag;
	double sum;
} t[N * 17];

inline void pushup(int o) {
	if (t[o].tag) return t[o].val = t[o].sum = 0, (void)0;
	t[o].val = t[t[o].lc].val + t[t[o].rc].val;
	t[o].sum = t[t[o].lc].sum + t[t[o].rc].sum;
}
inline void pushdown(int o) {
	if (!t[o].tag) return;
	if (t[o].lc) t[t[o].lc].tag = 1, t[t[o].lc].val = 0, t[t[o].lc].sum = 0;
	if (t[o].rc) t[t[o].rc].tag = 1, t[t[o].rc].val = 0, t[t[o].rc].sum = 0;
	t[o].tag = 0;
}
inline void qadd(int &o, int L, int R, int x, int k) {
	if (!k) return;
	if (!o) o = ++nod;
	if (L == R) return t[o].val += k, t[o].sum += k * log(L), (void)0;
	int M = (L + R) >> 1;
	pushdown(o);
	if (x <= M) qadd(t[o].lc, L, M, x, k);
	else qadd(t[o].rc, M + 1, R, x, k);
	pushup(o);
}
inline void qset(int o, int L, int R, int l, int r) {
	if (!o || !t[o].val || l > r) return;
	if (l <= L && R <= r) return t[o].tag = 1, t[o].val = 0, t[o].sum = 0, (void)0;
	int M = (L + R) >> 1;
	pushdown(o);
	if (l <= M) qset(t[o].lc, L, M, l, r);
	if (r > M) qset(t[o].rc, M + 1, R, l, r);
	pushup(o);
}
inline int qsum(int o, int L, int R, int l, int r) {
	if (!o || !t[o].val || l > r) return 0;
	if (l <= L && R <= r) return t[o].val;
	int M = (L + R) >> 1;
	pushdown(o);
	if (r <= M) return qsum(t[o].lc, L, M, l, r);
	if (l > M) return qsum(t[o].rc, M + 1, R, l, r);
	return qsum(t[o].lc, L, M, l, r) + qsum(t[o].rc, M + 1, R, l, r);
}
inline int qval(int o, int L, int R, int k) {
	if (L == R) return L;
	int M = (L + R) >> 1;
	pushdown(o);
	if (k <= t[t[o].lc].val) return qval(t[o].lc, L, M, k);
	else return qval(t[o].rc, M + 1, R, k - t[t[o].lc].val);
}
inline int merge(int o, int p) {
	if (!o || !p) return o ^ p;
	pushdown(o), pushdown(p);
	t[o].lc = merge(t[o].lc, t[p].lc);
	t[o].rc = merge(t[o].rc, t[p].rc);
	t[o].val += t[p].val, t[o].sum += t[p].sum;
	return o;
}
inline void newnode(int v) { qadd(rt[++n], 1, INF, v, 1), assert(t[rt[n]].val == 1); }

inline int find(int x) { return fa[x] == x ? x : fa[x] = find(fa[x]); }

inline void work() {
	while (m--) {
		int opt, x, y;
		read(opt);
		if (opt == 1) {
			read(x);
			newnode(x);
			fa[n] = n;
		} else if (opt == 2) {
			read(x), read(y);
			x = find(x), y = find(y);
			if (x == y) continue;
			fa[y] = x;
			rt[x] = merge(rt[x], rt[y]);
		} else if (opt == 3) {
			read(x), read(y);
			x = find(x);
			int cnt = qsum(rt[x], 1, INF, 1, y - 1);
			qset(rt[x], 1, INF, 1, y - 1);
			qadd(rt[x], 1, INF, y, cnt);
		} else if (opt == 4) {
			read(x), read(y);
			x = find(x);
			int cnt = qsum(rt[x], 1, INF, y + 1, INF);
			qset(rt[x], 1, INF, y + 1, INF);
			qadd(rt[x], 1, INF, y, cnt);
		} else if (opt == 5) {
			read(x), read(y);
			x = find(x);
			printf("%d\n", qval(rt[x], 1, INF, y));
		} else if (opt == 6) {
			read(x), read(y);
			x = find(x), y = find(y);
			if (t[rt[x]].sum > t[rt[y]].sum) puts("1");
			else puts("0");
		} else if (opt == 7) {
			read(x);
			x = find(x);
			printf("%d\n", t[rt[x]].val);
		}
	}
}

inline void init() {
	read(m);
}

int main() {
#ifdef hzhkk
	freopen("hkk.in", "r", stdin);
#endif
	init();
	work();
	fclose(stdin), fclose(stdout);
	return 0;
}
posted @ 2019-10-14 22:29  hankeke303  阅读(202)  评论(0编辑  收藏  举报