一开始想的是分类讨论,看那条边加了之后成不成环,如果不成环且权值在前n-1,则一定在MST里,不在前n-1则不在MST里;如果成环了,如果权值不在前n-1,则不在MST里,如果权值在前n-1,则需要判断环中其余边是不是都<wi,如果是,则在MST里面,否则就不在
然后问题就在于,怎么看环中其余边是不是都<wi
有两个想法:
一个是dfs遍历环统计最大值,但是考虑到最坏情况复杂度为NQ,因此不可能这么做
一个是维护树链RMQ,可能要用树剖来写,但是我不会QwQ
事实上这也是做法之一,但是不需要这么麻烦
题意:
有一个图,它有MST,对于每次询问都问是否能加一条边,使得该边在新图的MST中,如果是输出Yes,否则就是No
思路:
考虑将原边和查询边一起参与Kruskal的过程(其实就是将询问的边按边权离线,这样更容易地用DS维护)
如果是原边,则直接加入
如果是查询边,则直接看是否成环即可
为什么,因为它是从小到大加边的,所以此时如果成环了,环内其余边都<=wi,就不需要树链RMQ了
感觉将查询边一起参与Kruskal的过程很帅啊
Code:
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mxn=2e5+10;
const int mxe=2e5+10;
struct ty{
int u,v,w,id;
}e[mxe<<2];
int N,M,Q,a,b,c;
int f[mxn],ans[mxn];
int find(int x){
return f[x]=(x==f[x])?x:find(f[x]);
}
void join(int u,int v){
int f1=find(u),f2=find(v);
if(f1!=f2) f[f1]=f2;
}
bool cmp(ty x,ty y){
return x.w<y.w;
}
void solve(){
cin>>N>>M>>Q;
for(int i=1;i<=N;i++) f[i]=i;
for(int i=1;i<=M;i++){
cin>>a>>b>>c;
e[i]={a,b,c,0};
}
for(int i=1;i<=Q;i++){
cin>>a>>b>>c;
e[M+i]={a,b,c,i};
}
sort(e+1,e+1+M+Q,cmp);
for(int i=1;i<=M+Q;i++){
if(e[i].id==0) join(e[i].u,e[i].v);
else{
if(find(e[i].u)!=find(e[i].v)) ans[e[i].id]=1;
else ans[e[i].id]=0;
}
}
for(int i=1;i<=Q;i++){
if(ans[i]) cout<<"Yes"<<'\n';
else cout<<"No"<<'\n';
}
}
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int __=1;//cin>>__;
while(__--)solve();return 0;
}
2023.08.11 upd:
新版离线写法,更高级更好用:
#include <bits/stdc++.h>
using i64 = long long;
constexpr int N = 2e5 + 10;
constexpr int M = 2e5 + 10;
constexpr int mod = 998244353;
std::vector<int> f(N), ans(N);
int find(int x) {
return f[x] = (x == f[x]) ? x : find(f[x]);
}
void join(int u, int v) {
int f1 = find(u), f2 = find(v);
if (f1 != f2) {
f[f1] = f2;
}
}
void solve() {
int n, m, q;
std::cin >> n >> m >> q;
for (int i = 0; i < n; i ++) {
f[i] = i;
}
std::vector<std::array<int,3> > e;
for (int i = 0; i < m; i ++) {
int u, v, w;
u --, v --;
std::cin >> u >> v >> w;
e.push_back({w, u, v});
}
std::sort(e.begin(), e.end());
std::vector<std::array<int,4> > Q;
for (int i = 0; i < q; i ++) {
int u, v, w;
std::cin >> u >> v >> w;
Q.push_back({w, u, v, i});
}
std::sort(Q.begin(), Q.end());
int idx = 0;
for (int i = 0; i < m; i ++) {
while(idx < q && Q[idx][0] <= e[i][0]) {
if (find(Q[idx][1]) == find(Q[idx][2])) {
ans[Q[idx][3]] = 0;
}else {
ans[Q[idx][3]] = 1;
}
idx ++;
}
join(e[i][1], e[i][2]);
}
for (int i = 0; i < q; i ++) {
if (ans[i] == 1) {
std::cout << "Yes" << "\n";
}else {
std::cout << "No" << "\n";
}
}
}
signed main() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
int t = 1;
while(t --) {
solve();
}
return 0;
}