社区讨论
kruskal重构树 马蜂优良TLE 0
P1967[NOIP 2013 提高组] 货车运输参与者 3已保存回复 3
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 3 条
- 当前快照
- 1 份
- 快照标识符
- @lxlx3nee
- 此快照首次捕获于
- 2024/06/19 22:17 2 年前
- 此快照最后确认于
- 2024/06/20 13:37 2 年前
所有的数据点都 TLE , 下载小数据后在本地跑的飞快, 不理解怎么 TLE 的
CPP#include<algorithm>
#include<iostream>
#include<iomanip>
#include<cstring>
using namespace std;
const int N = 5e4+10;
int n, m, tot, h[N], dsu[N], val[N];
int dep[N], siz[N], son[N], top[N], fa[N];
int idx; struct Forward_star{int to, ne;} e[N];
struct Edges{int u, v, w;} g[N];
bool cmp(Edges x, Edges y) {return x.w > y.w;}
void add_edge(int u, int v) {
e[++ idx].to = v;
e[idx].ne = h[u];
h[u] = idx;
}
int find(int x) {
if(x != dsu[x]) dsu[x] = find(dsu[x]);
return dsu[x];
}
int dfs1(int u) {
siz[u] = 1;
son[u] = -1;
for(int i = h[u]; ~i; i = e[i].ne) {
int v = e[i].to;
if(dep[v]) continue;
fa[v] = u;
dep[v] = dep[u] + 1;
dfs1(v);
siz[u] += siz[v];
if(!~son[u] || siz[son[u]] < siz[v])
son[u] = v;
}
}
void dfs2(int u, int t) {
top[u] = t;
if(~son[u]) dfs2(son[u], t);
for(int i = h[u]; ~i; i = e[i].ne)
if(e[i].to != son[u] && e[i].to != fa[u])
dfs2(e[i].to, e[i].to);
}
int lca(int u, int v) {
while(top[u] != top[v]) {
if(dep[top[u]] < dep[top[v]]) swap(u, v);
u = fa[top[u]];
}
return dep[u] > dep[v] ? v : u;
}
inline void kruskal() {
sort(g + 1, g + m + 1, cmp);
for(int i = 1; i <= m; i ++) {
int u = find(g[i].u), v = find(g[i].v);
if(u != v) {
val[++ tot] = g[i].w;
dsu[u] = dsu[v] = dsu[tot] = tot;
add_edge(u, tot);
add_edge(tot, u);
add_edge(v, tot);
add_edge(tot, v);
}
}
for(int i = 1; i <= n; i ++) {
if(dep[i]) continue;
int f = find(i);
dep[f] = 1;
dfs1(f);
dfs2(f, f);
}
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0); cout.tie(0);
memset(h, -1, sizeof h);
cin >> n >> m; tot = n;
for(int i = 1; i <= n; i ++) dsu[i] = i;
for(int i = 1; i <= m; i ++)
cin >> g[i].u >> g[i].v >> g[i].w;
kruskal();
int q, u, v;
cin >> q;
// for(int i = 1; i <= tot; i ++)
// cout << dep[i] << ' ';
while(q --) {
cin >> u >> v;
if(find(u) != find(v)) cout << "-1\n";
else cout << val[lca(u, v)] << '\n';
}
}
回复
共 3 条回复,欢迎继续交流。
正在加载回复...