牛客练习赛71 F 红蓝图（kruskal重构树）

#include <bits/stdc++.h>using namespace std;const int N = 4e5 + 10;struct Res {int u, v, w, op;bool operator < (const Res &t) const {return w < t.w;}
}edge[N];int n, m, Q, x[N], y[N];int root[N], ls[N << 5], rs[N<< 5], sum[N << 5], num;void update(int &rt, int pre, int l, int r, int x, int v) {rt = ++num;ls[rt] = ls[pre], rs[rt] = rs[pre], sum[rt] = sum[pre] + v;if (l == r) {return ;}int mid = l + r >> 1;if (x <= mid) {update(ls[rt], ls[pre], l, mid, x, v);}else {update(rs[rt], rs[pre], mid + 1, r, x, v);}
}int query(int rt1, int rt2, int l, int r, int L, int R) {if (l >= L && r <= R) {return sum[rt2] - sum[rt1];}int ans = 0, mid = l + r >> 1;if (L <= mid) {ans += query(ls[rt1], ls[rt2], l, mid, L, R);}if (R > mid) {ans += query(rs[rt1], rs[rt2], mid + 1, r, L, R);}return ans;
}struct Kruskal {int head[N], to[N], nex[N], cnt = 1;int ff[N], value[N], fa[N][21], l[N], r[N], rk[N], tot, nn;void add(int x, int y) {to[cnt] = y;nex[cnt] = head[x];head[x] = cnt++;}int find(int rt) {return ff[rt] == rt ? rt : ff[rt] = find(ff[rt]);}void dfs(int rt, int f) {fa[rt][0] = f, l[rt] = ++tot, rk[tot] = rt;for (int i = 1; i <= 20; i++) {fa[rt][i] = fa[fa[rt][i - 1]][i - 1];}for (int i = head[rt]; i; i = nex[i]) {if (to[i] == f) {continue;}dfs(to[i], rt);}r[rt] = tot;}void kruskal(int rev) {if (rev) {reverse(edge + 1, edge + 1 + m);}for (int i = 1; i < N; i++) {ff[i] = i;}nn = n;for (int i = 1, cur = 1; i <= m && cur < n; i++) {if (edge[i].op != rev) {continue;}int u = find(edge[i].u), v = find(edge[i].v);if (u != v) {cur++, nn++;ff[u] = ff[v] = nn;value[nn] = edge[i].w;add(nn, u), add(nn, v);if (u <= n) {value[u] = edge[i].w;}if (v <= n) {value[v] = edge[i].w;}}}for (int i = 1; i < 2 * n; i++) {if (ff[i] == i) {dfs(i, 0);}}}
}a, b;int main() {// freopen("in.txt", "r", stdin);// freopen("out.txt", "w", stdout);// ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);scanf("%d %d %d", &n, &m, &Q);for (int i = 1, u, v, op; i <= m; i++) {scanf("%d %d %d", &u, &v, &op);u++, v++;edge[i] = {u, v, i, op};}sort(edge + 1, edge + 1 + m);a.kruskal(0);b.kruskal(1);for (int i = 1; i < 2 * n; i++) {x[i] = a.rk[i];y[i] = b.l[i];}for (int i = 1; i < 2 * n; i++) {root[i] = root[i - 1];if (x[i] <= n) {update(root[i], root[i - 1], 1, 2 * n, y[x[i]], 1);}}int x, t;while (Q--) {scanf("%d %d", &x, &t);int u = x + 1, v = x + 1;for (int i = 20; i >= 0; i--) {if (a.fa[u][i] && a.value[a.fa[u][i]] <= t) {u = a.fa[u][i];}}for (int i = 20; i >= 0; i--) {if (b.fa[v][i] && b.value[b.fa[v][i]] >= t) {v = b.fa[v][i];}}int l1 = a.l[u], r1 = a.r[u], l2 = b.l[v], r2 = b.r[v];printf("%d\n", query(root[l1 - 1], root[r1], 1, 2 * n, l2, r2));}return 0;
}

牛客练习赛71 F 红蓝图（kruskal重构树）相关推荐

解题报告（一）C、（牛客练习赛41 F）简单数学题（数论 + FWT）（3.5）
繁凡出品的全新系列:解题报告系列 -- 超高质量算法题单,配套我写的超高质量题解和代码,题目难度不一定按照题号排序,我会在每道题后面加上题目难度指数(1∼51 \sim 51∼5),以模板题难度 11 ...
AGC002（D~F）【Kruskal重构树,博弈论,dp】
正题 AT1998 [AGC002D] Stamp Rally[Kruskal重构树,倍增] https://www.luogu.com.cn/problem/AT1998 题目大意给出nnn个点m ...
牛客练习赛29: F. 算式子
链接:https://www.nowcoder.com/acm/contest/211/F 来源:牛客网题目描述给定个整数 .保证 . 对于每个 ,求出 .为了避免过量输出,你只需要将所 ...
牛客练习赛50 F tokitsukaze and Another Protoss and Zerg（分治NTT）
一些废话: 写完上一题(分治FFT)之后记起之前牛客写的这道题,题解是用启发式NTT,每次选两个项数最小的合并,当时没有整理NTT的模板所以就先mark了,今天受到分治FFT的启发,想用分治NTT来解 ...
牛客练习赛 43 F Tachibana Kanade Loves Game 容斥原理(Java版ac)
链接:https://ac.nowcoder.com/acm/contest/548/F 来源:牛客网题目描述立华奏是一个天天打比赛的萌新. 省选将至,萌新立华奏深知自己没有希望进入省队,因此开始 ...
牛客练习赛63 F 牛牛的树行棋（SG函数+树差分）
链接:https://ac.nowcoder.com/acm/contest/5531/F 来源:牛客网牛牛的树行棋时间限制:C/C++ 2秒,其他语言4秒空间限制:C/C++ 524288K, ...
【牛客网】牛客练习赛19 F 算式子【数学--递推、前缀、数字】
传送门:算式子花了一些时间理解AC的代码,震惊,代码真的是短小精悍,推理能力很强亦或者是做题多,见的多. 能够理解里面的逻辑真的挺难的题意给定n,m,\(1\le x\le m\),求\(\su ...
牛客练习赛50 F tokitsukaze and Another Protoss and Zerg
tokitsukaze and Another Protoss and Zerg 考虑生成函数,每一场的生成函数为∑j=1b[i]Cb[i]j+∑j=1a[i]Ca[i]jxj\sum\limits_ ...
牛客练习赛 71 AC
A - 回文数回文数条件:奇数个数的个数最多有一个如果0的数量是0,那么直接先找到奇数个数的数是谁,然后正序输出一半逆序输出一半即可. 如果0的数量不是0,只需要找到一个偶数个个数的数放置第一个即 ...

牛客练习赛71 F 红蓝图（kruskal重构树）

红蓝图

牛客练习赛71 F 红蓝图（kruskal重构树）相关推荐

最新文章

热门文章