湘南学院ACM暑期排位赛第二场题解报告

D - Shortest Path 3

题目大意：

给你一个无向图，图中有 $n$ 个顶点和 $m$ 条边。每个顶点 $i(1\le i\le n)$ 的权重为$a_i$。每条边 $j(1\le j\le m)$ 双向连接顶点 $u_j$ 和 $v_j$ ，权重为 $b_j$。

该图中路径的权重定义为路径上出现的顶点和边的权重之和。

针对每个 $i=2,3,\dots ,n$ 求解以下问题：

求从顶点 $1$ 到顶点 $i$ 的路径的最小权重。

数据范围

• $2\le n\le 2\times 10^5$
• $n-1\le m\le 2\times 10^5$
• $1\le u_j<v_j\le n$
• $(u_i,v_i)\ne (u_j,v_j)$ 如果 $i\ne j$ .
• 图是连通的。
• $0\le a_i\le 10^9$
• $0\le b_j\le 10^9$
• 所有输入值均为整数。

分析：

很明显可以看出来，这是从起点到每个点的最短距离，则这一题可以使用$Dijkstra$算法求解，这里$Dijkstra$算法是每个人都必须要掌握的板子，我便不再赘述了。要注意的是，朴素版本的$Dijkstra$算法时间复杂度为$O(n^2)$的，其中$n$为点的数量。这里因为数据范围较大，所以需要使用堆优化版的，即优先队列$Dijkstra$，时间复杂度为$O(m\times log(n))$，其中$m$为边的数量，$n$为点数。

当你确定算法后，就可以发现，其实这就是一道很简单的板子题，唯一不同的是：在$Dijkstra$算法中并不存在点权这一说。如下图:

设·从 $1$ 号点到 $5$ 号点的最短距离就是上图中的路线，其中$w,x,y,z$为边权，$a_1,a_2,a_3,a_4,a_5$为点权。

纯板子$Dijkstra$中，$dis{t}_5^{\prime }=w+x+y+z$，但是本题目中，我们需要求得的应该为$dis{t}_5=a_1+a_2+a_3+a_4+a_5+w+x+y+z$。由于$Dijkstra$的特性，我们仅能知道该最短路的起点和终点，无法得出此最短路中经过了哪些点，要加点权的话，我们不妨可以先将点权集成到边权中。 如下图：

此时我们通过$Dijkstra$计算出来的$dis{t}_5$为：
$$\begin{array}{rl}dis{t}_5^{\prime }& =(a_1+w+a_2)+(a_2+x+a_3)+(a_3+y+a_4)+(a_4+z+a_5)\\ & =a_1+2\ast a_2+2\ast a_3+2\ast a_4+a_5+w+x+y+z\end{array}$$
观察上式，发现这与我们所求的$dis{t}_5$相比，中间每个点的点权我们都多加了一遍。上述加粗部分所说，由于我们不知道最短路的过程中会经过哪些点，所以我们无法将多余的部分删除，但是我们却是知道起点和终点，所以我们将起点和终点的点权再加一遍就会得到：
$$dis{t}_5^{\prime }+a_1+a_5=2\ast a_1+2\ast a_2+2\ast a_3+2\ast a_4+2\ast a_5+w+x+y+z$$
可以发现，我们该式与我们所求的$dis{t}_5$相比，点权部分变成了两倍，边权部分正常，由于我们点权和边权都加到了一起，无法区分开来，那我们索性再加一遍边权，我们将一条边的边权值在改变一下：

此时我们纯$Dijkstra$板子所求$dis{t}_5^{\prime }$为:
$$\begin{array}{rl}dis{t}_5^{\prime }+a_1+a_5& =(a_1+2\ast w+a_2)+(a_2+2\ast x+a_3)+(a_3+2\ast y+a_4)+(a_4+2\ast z+a_5)+a_1+a_5\\ & =2\ast a_1+2\ast a_2+2\ast a_3+2\ast a_4+2\ast a_5+2\ast w+2\ast x+2\ast y+2\ast z\\ & =2\ast (a_1+a_2+a_3+a_4+a_5+w+x+y+z)\\ & =2\ast dis{t}_5\end{array}$$
上式中$dis{t}_5$为该题所求。

AC代码：

#include <bits/stdc++.h>
#define _1 first
#define _2 second
using i64 = long long;

const int N = 1e6 + 5, M = 1e2 + 5;
const i64 INF = 1e18;

int n, m, k;
int h[N], cnt;
i64 a[N], dist[N];

struct Edge {
    i64 v, w, next;
} g[N];

inline void add(int u, int v, i64 w) {
    cnt ++;
    g[cnt].w = w;
    g[cnt].v = v;
    g[cnt].next = h[u];
    h[u] = cnt;
}

struct node {
    i64 d, u;
    bool operator<(const node& t)const {
        return d > t.d;
    } // 由于使用的优先队列priority_queue默认为大根堆，这里重载一下
};

inline void dijkstra() {	// dijkstra纯板子
    int s = 1;
    for (int i = 1; i <= n; i ++)   dist[i] = INF;
    dist[s] = 0;
    std::priority_queue<node> pq;
    
    pq.push(node{0, s});
    while (!pq.empty()) {
        node tmp = pq.top();    pq.pop();
        i64 u = tmp.u, d = tmp.d;
        if (d != dist[u])   continue ;
        for (int i = h[u]; i; i = g[i].next) {
            i64 v = g[i].v, w = g[i].w;
            if (dist[u] + w < dist[v]) {
                dist[v] = dist[u] + w;
                pq.push(node{dist[v], v});
            }
        }
    }
    return ;
}

void solve()
{
    std::cin >> n >> m;
    
    for (int i = 1; i <= n; i ++) {
        std::cin >> a[i];
    }
    
    for (int i = 1; i <= m; i ++) {
        int x, y, z;
        std::cin >> x >> y >> z;
        add(x, y, z + z + a[x] + a[y]);
        add(y, x, z + z + a[x] + a[y]);
        	// 无向图，定义边权 = 2倍原本边权 + 所连两点的点权
    }
    dijkstra();
    for (int i = 2; i <= n; i ++) {
        std::cout << (dist[i] + a[1] + a[i]) / 2 << ' ';
    } std::cout << '\n';
    
}

int main()
{
    std::ios::sync_with_stdio(false);std::cout.tie(nullptr);std::cout.tie(nullptr);
    int _ = 1;  // std::cin >> (_);
    while (_ --) {
        solve();
    }
    return 0;
}