1. 程式人生 > >CSP 地鐵修建 最小生成樹+並查集

CSP 地鐵修建 最小生成樹+並查集

printf col 情況 cnblogs con 1.0 nbsp width int

地鐵修建

試題編號: 201703-4
試題名稱: 地鐵修建
時間限制: 1.0s
內存限制: 256.0MB
問題描述: 問題描述   A市有n個交通樞紐,其中1號和n號非常重要,為了加強運輸能力,A市決定在1號到n號樞紐間修建一條地鐵。
  地鐵由很多段隧道組成,每段隧道連接兩個交通樞紐。經過勘探,有m段隧道作為候選,兩個交通樞紐之間最多只有一條候選的隧道,沒有隧道兩端連接著同一個交通樞紐。
  現在有n家隧道施工的公司,每段候選的隧道只能由一個公司施工,每家公司施工需要的天數一致。而每家公司最多只能修建一條候選隧道。所有公司同時開始施工。
  作為項目負責人,你獲得了候選隧道的信息,現在你可以按自己的想法選擇一部分隧道進行施工,請問修建整條地鐵最少需要多少天。 輸入格式   輸入的第一行包含兩個整數n, m,用一個空格分隔,分別表示交通樞紐的數量和候選隧道的數量。
  第2行到第m+1行,每行包含三個整數a, b, c,表示樞紐a和樞紐b之間可以修建一條隧道,需要的時間為c天。 輸出格式   輸出一個整數,修建整條地鐵線路最少需要的天數。 樣例輸入 6 6
1 2 4
2 3 4
3 6 7
1 4 2
4 5 5
5 6 6 樣例輸出 6 樣例說明   可以修建的線路有兩種。
  第一種經過的樞紐依次為1, 2, 3, 6,所需要的時間分別是4, 4, 7,則整條地鐵線需要7天修完;
  第二種經過的樞紐依次為1, 4, 5, 6,所需要的時間分別是2, 5, 6,則整條地鐵線需要6天修完。
  第二種方案所用的天數更少。 評測用例規模與約定   對於20%的評測用例,1 ≤ n ≤ 10,1 ≤ m ≤ 20;
  對於40%的評測用例,1 ≤ n ≤ 100,1 ≤ m ≤ 1000;
  對於60%的評測用例,1 ≤ n ≤ 1000,1 ≤ m ≤ 10000,1 ≤ c ≤ 1000;
  對於80%的評測用例,1 ≤ n ≤ 10000,1 ≤ m ≤ 100000;
  對於100%的評測用例,1 ≤ n ≤ 100000,1 ≤ m
≤ 200000,1 ≤ a, bn,1 ≤ c ≤ 1000000。

  所有評測用例保證在所有候選隧道都修通時1號樞紐可以通過隧道到達其他所有樞紐。
求連通路徑中天數最大值最小情況。
#include<stdio.h>
#include<algorithm>
using namespace std;

int f[100005];

struct Edge{
    int u,v,w;
}edge[200005];

bool cmp(Edge a,Edge b)
{
    return a.w<b.w;
}

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

int kruskal(int m,int n)
{
    int i;
    for(i=1;i<=n;i++){
        f[i]=i;
    }
    sort(edge+1,edge+m+1,cmp);
    for(i=1;i<=m;i++){
        int u=edge[i].u;
        int v=edge[i].v;
        int w=edge[i].w;
        int fu=find(u),fv=find(v);
        if(fu!=fv) f[fv]=fu;
        if(find(1)==find(n)) return w;
    }
}

int main()
{
    int n,m,u,v,w,i;
    scanf("%d%d",&n,&m);
    for(i=1;i<=m;i++){
        scanf("%d%d%d",&u,&v,&w);
        edge[i].u=u;
        edge[i].v=v;
        edge[i].w=w;
    }
    printf("%d\n",kruskal(m,n));
    return 0;
}

CSP 地鐵修建 最小生成樹+並查集