CodeChef：Two Closest（最短路 & 思维）

最新推荐文章于 2021-09-18 10:37:15 发布

原创最新推荐文章于 2021-09-18 10:37:15 发布 · 488 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

最短路/差分约束专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种算法，用于在一个加权图中找到任意两个特殊节点间的最短距离。通过使用Dijkstra算法并针对特殊节点进行优化，该方法能够在满足特定约束条件下高效地解决问题。

You are given a weighted graph with N nodes and M edges. Some of the nodes are marked as special nodes. Your task is to find the shortest pairwise distance between any two different special nodes.

Input

The first line of the input contains three space-separated integers N, M and K denoting the number of nodes, the number of edges, and the number of special nodes.

The following line contains K space-separated distinct integers A₁, A₂, ..., A_K , denoting the special nodes.

Each of the following M lines (say, the j^th) contains a triple X_j Y_j Z_j, denoting the edge connecting the nodes X_j and Y_j, and having the weight of Z_j.

Output

Output the shortest pairwise distance between any two different special nodes.

Constraints

2 ≤ K ≤ N
The given graph is connected.
The given graph doesn't contain self loops and multiple edges.
1 ≤ A_i ≤ N
1 ≤ Z_j ≤ 10⁴
1 ≤ X_j, Y_j ≤ N

Subtasks

Subtask #1 (20 points): 2 ≤ N ≤ 300, N-1 ≤ M ≤ N*(N-1)/2, 2 ≤ K ≤ N
Subtask #2 (25 points): 2 ≤ N ≤ 10⁵, N-1 ≤ M ≤ 10⁵, 2 ≤ K ≤ 10
Subtask #3 (55 points): 2 ≤ N ≤ 10⁵, N-1 ≤ M ≤ 3 * 10⁵, 2 ≤ K ≤ 10⁴

Example

Input:
5 5 3
1 3 5
1 2 3
2 3 4
3 4 1
4 5 8
1 5 19

Output:
7

Explanation

Nodes 1, 3 and 5 are special nodes. Shortest distance between nodes 1 and 3 is 7 and that between nodes 3 and 5 is 9. Shortest distance between nodes 1 and 5 is 16. Minimum of these distances is 7. Hence answer is 7.

题意：给一个无向图，K个特殊点，问这K个点中最近的两个点的距离。

思路：特殊点任取一个作为起点跑dijk遇到另外的特殊点就结束，更新最短距离imin，从下一个特殊点开始继续跑，超过imin就break，显然复杂度O(NlogN)。

# include <bits/stdc++.h>
# define pb push_back
# define mp make_pair
# define pii pair<int,int>
# define A first
# define B second
using namespace std;
const int maxn = 1e5+10;
const int INF = 0x3f3f3f3f;
int a[maxn], vis[maxn], dis[maxn];
vector<pii >g[maxn];
priority_queue<pii, vector<pii >, greater<pii > >q;
int main()
{
    int t, n, m, k, u, v, w, cnt, imin = INF;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=0; i<k; ++i)
    {
        scanf("%d",&t);
        a[t] = 1;
    }
    for(int i=0; i<m; ++i)
    {
        scanf("%d%d%d",&u,&v,&w);
        g[u].pb(mp(v,w));
        g[v].pb(mp(u,w));
    }
    for(int i=1; i<=n; ++i)
    {
        if(!a[i]) continue;
        for(int j=1; j<=n; ++j) dis[j] = INF, vis[j] = 0;
        cnt = dis[i] = 0;
        while(!q.empty()) q.pop();
        q.push(mp(0, i));
        while(!q.empty())
        {
            int u = q.top().B;
            q.pop();
            vis[u] = 1;
            cnt += a[u];
            if(dis[u] > imin) break;
            if(cnt == 2)
            {
                imin = min(imin, dis[u]);
                break;
            }
            for(auto j : g[u])
            {
                int v = j.A, w = j.B;
                if(vis[v]) continue;
                if(dis[v] > dis[u]+w)
                {
                    dis[v] = dis[u]+w;
                    q.push(mp(dis[v], v));
                }
            }
        }
    }
    printf("%d\n",imin);
    return 0;
}