P1396 营救 /二分/并查集/最小生成树

营救 - 洛谷二分答案/并查集处理

 

题意：题目的问法，最大值最小的情况。很容易想到二分，而要让值最小同时肯定也必须到达目的地。所以二分答案，同时并查集合并比这个值小的所有顶点，最后判断目标起点和终点有没有联通调整二分范围。
 

#include <iostream>
#include<vector>
using namespace std;
const int N= 2e4  +10;
int pre[N];
int fnd(int a){
    return pre[a]==a?a:pre[a]=fnd(pre[a]);
}
void join(int a,int b){
    int t1=fnd(a),t2=fnd(b);
    if(t1!=t2)
    pre[t1]=t2;
}
int n,m,s,t;
int u[N],v[N],w[N];
int vis[N];
int gra[N][N];
int judge(int a,int b){//判断是否联通
    return fnd(a)==fnd(b);
}
int check(int k){
    for(int j=1;j<=n;j++){
        pre[j]=j;
    }
    for(int j=1;j<=m;j++){
        if(w[j]<=k){
            join(u[j],v[j]);
        }
    }
    return judge(s,t);
}
int main()
{
    cin>>n>>m>>s>>t;
    for(int j=1;j<=m;j++){
        int U,V,W;
        cin>>U>>V>>W;
        u[j]=U;
        v[j]=V;
        w[j]=W;
 
    }
    int l=-1,r=10001;
    int mid;
    while(l+1!=r){
      mid=(l+r)>>1;
      if(check(mid)){
          r=mid;
      }
      else
          l=mid;
    }
    cout<<r<<endl;
    return 0;
}

还有一种做法就是最小生成树，因为要求树的最大边权最小，所以枚举最大边权，然后逐个判断合并顶点，是否可以抵达终点。这个做法就很类似kruscal算法。

//#include<bits/std c++.h>
#include <iostream>
#include<ctime>
#include<math.h>
#include<string>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<cstring>
#include<algorithm>
#include<limits.h>
#include<unordered_map>
#include<unordered_set>
#include<regex>
using namespace std;

#define f first
#define s second
#define ll long long
#define int ll
#pragma warning (disable:4996);
const int N = 2e5 + 10;
int T, n, m, q;
int pre[N];
struct node {//存储边
    int from;
    int to;
    int pw;
    int operator<(node& a) {
        return pw < a.pw;
    }
}edg[N];
void init(int n) {
    for (int j = 1; j <= n; j++) {
        pre[j] = j;
    }
}
inline int fnd(int k) {
    return pre[k] == k ? k : pre[k] = fnd(pre[k]);
}
void join(int a, int b) {
    pre[fnd(b)] = fnd(a);
}
signed main()
{
    ios::sync_with_stdio(false); cout.tie(NULL);
#ifndef ONLINE_JUDGE
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
#endif // !ONLINE_JUDGE
    int s, t;
    cin >> n >> m>>s>>t;
    for (int j = 1; j <= m; j++) {
        int a, b, c;
        cin >> a >> b >> c;
        edg[j] = { a,b,c };
    }
    init(n);
    int cnt = 0;//计算链接边数
    //n个顶点最多（无环的时候）产生n-1个边
    int ok = 0;
    int sum = 0;
    sort(edg + 1, edg + 1 + m);
    for (int j = 1; j <= m; j++) {
        if (fnd(edg[j].from) != fnd(edg[j].to)) {
            pre[fnd(edg[j].from)] = fnd(edg[j].to);
            if (fnd(s) == fnd(t)) {
                cout << edg[j].pw << endl;
                break;
            }//起点终点联通了
        }
    }
 
}