CCF201703-4 地铁修建(最短路 / 最小生成树+并查集)

问题描述

试题编号：	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, b ≤ n，1 ≤ c ≤ 1000000。 　　所有评测用例保证在所有候选隧道都修通时1号枢纽可以通过隧道到达其他所有枢纽。

---

两种做法。分别为最短路与最小生成树。

最短路：答案为所有可能的线路中用时最短的一个，而所用最少天数取决于一条线路中所需修建时间最长的一条隧道，所以这是一个在最大值中找最小值的问题。跑一遍dijkstra或者spfa，对于一条边u->v，更新dis[v]时，先比较前驱的最短路径dis[u]与u->v的权值，取较大者为tmpdis，再将tmpdis与dis[v]比较，令dis[v]取较小者，就完成了更新。

需要注意的是，最短路的做法需要关同步，否则会超时；最小生成树不需要关同步。

下面是dijkstra的代码：

#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e5+10;
const int maxm = 2e5+10;
struct node
{
    int x, d;
    node() {}
    node(int X, int D): x(X), d(D) {}
    bool operator < (const node& a) const
    {
        if(d == a.d)
            return x < a.x;
        return d > a.d;
    }
};
int n, m;
int dis[maxn];
vector <node> r[maxn];

void read()
{
    cin >> n >> m;
    for(int i = 0; i < m; ++i)
    {
        int u, v, w;
        cin >> u >> v >> w;
        r[u].push_back(node(v, w));
        r[v].push_back(node(u, w));
    }
}

void dijkstra(int s)
{
    for(int i = 2; i <= n; ++i)
        dis[i] = INF;
    dis[1] = 0;
    int vis[maxn] = {0};
    priority_queue <node> q;
    q.push(node(s, dis[s]));

    while(!q.empty())
    {
        node cur = q.top();
        q.pop();
        if(vis[cur.x])
            continue;
        for(int i = 0; i < r[cur.x].size(); ++i)
        {
            node next = r[cur.x][i];
            int tmpdis = max(dis[cur.x], next.d);
            dis[next.x] = min(tmpdis, dis[next.x]);
            q.push(node(next.x, dis[next.x]));
        }
        vis[cur.x] = 1;
    }
}
void solve()
{
    dijkstra(1);
    cout << dis[n];
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    read();
    solve();
    return 0;
}

最短路spfa版本：

#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e5+10;
const int maxm = 2e5+10;
int n, m;
int dis[maxn];
vector <int> r[maxn];
vector <int> w[maxn]; //n的最大值为1e5，无法使用邻接矩阵

void read()
{
    cin >> n >> m;
    for(int i = 0; i < m; ++i)
    {
        int u, v, l;
        cin >> u >> v >> l;
        r[u].push_back(v);
        r[v].push_back(u);
        w[u].push_back(l);
        w[v].push_back(l);
    }
}

int vis[maxn];
void spfa(int s)
{
    for(int i = 2; i <= n; ++i)
        dis[i] = INF;
    dis[1] = 0;
    queue <int> q;
    q.push(s);
    vis[s] = 1;
    while(!q.empty())
    {
        int cur = q.front();
        q.pop();
        vis[cur] = 0;
        for(int i = 0; i < r[cur].size(); ++i)
        {
            int next = r[cur][i], W = w[cur][i];
            int tmpdis = max(dis[cur], W);
            if(tmpdis < dis[next])
            {
                dis[next] = tmpdis;
                if(!vis[next])
                {
                    q.push(next);
                    vis[next] = 1;
                }
            }
        }
    }
}

void solve()
{
    spfa(1);
    cout << dis[n];
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    read();
    solve();
    return 0;
}

最小生成树+并查集做法。由于同时开工，所以题意可理解为求最小生成树中的最大边权。于是，可以将所有边按权值升序排序，利用Kruskal算法，依次对边的两点进行并查集的合并操作，直到节点1与节点n同根，即起点与终点连通时，当前正在操作的边的权值即为答案。

#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e5+10;
const int maxm = 2e5+10;
struct Edge
{
    int u, v, w;
    bool operator < (const Edge& e) const
    {
        return w < e.w;
    }
}edge[maxm];
int n, m;
int p[maxn];

void read()
{
    cin >> n >> m;
    for(int i = 0; i < m; ++i)
        cin >> edge[i].u >> edge[i].v >> edge[i].w;
}

int find(int x)
{
    int r = x;
    while(p[r] != r)
        r = p[r];

    int i = x, j;
    while(i != r) //路径压缩
    {
        j = p[i];
        p[i] = r;
        i = j;
    }
    return r;
}

void Union(int x, int y)
{
    int rx = find(x), ry = find(y);
    if(rx != ry)
        p[rx] = p[ry];
}

void solve()
{
    sort(edge, edge+m);
    for(int i = 0; i <= n; ++i)
        p[i] = i;
    for(int i = 0; i < m; ++i)
    {
        Union(edge[i].u, edge[i].v);
        if(find(1) == find(n)) //1与n同根，则连通
        {
            cout << edge[i].w;
            break;
        }
    }
}

int main()
{
    read();
    solve();
    return 0;
}