BZOJ1576: [Usaco2009 Jan]安全路经Travel(最短路 并查集)

#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#include<algorithm>
#define Pair pair<int, int> 
#define MP(x, y) make_pair(x, y)
#define fi first
#define se second
using namespace std;
const int MAXN = 1e6 + 10, INF = 1e9 + 10;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int N, M;
vector<Pair> v[MAXN];
struct Edge {
    int u, v, w;
}E[MAXN];
int num = 0;
int dis[MAXN], top[MAXN], vis[MAXN], cnt = 0, ans[MAXN];
void Dij() {
    memset(dis, 0x7f, sizeof(dis));
    dis[1] = 0;
    priority_queue<Pair> q; q.push(MP(0, 1));
    while(!q.empty()) {
        int p = q.top().se; q.pop();
        if(vis[p]) continue; vis[p] = 1;
        for(int i = 0; i < v[p].size(); i++) {
            int to = v[p][i].fi, w = v[p][i].se;
            if(dis[to] > dis[p] + w && (!vis[to])) {
                top[to] = p;
                dis[to] = dis[p] + w;
                q.push(MP(-dis[to], to));
            }
        }
    }
}
int comp(const Edge &a, const Edge &b) {
    return dis[a.u] + dis[a.v] + a.w < dis[b.u] + dis[b.v] + b.w;
}
int fa[MAXN];
int unionn(int x, int y) {
    fa[x] = y;
}
int Find(int x) {
    if(fa[x] == x) return fa[x];
    else return fa[x] = Find(fa[x]);
}
int solve(int x, int y, int w) {
    while((x = Find(x)) != (y = Find(y))) {
        //int dx = dis[x], dy = dis[y];
        if(dis[x] < dis[y]) swap(x, y);
        ans[x] = w - dis[x];
        x = (fa[x] = top[x]);
        cnt++;
    }
}
int main() {
    N = read(); M = read();    
    for(int i = 1; i <= M; i++) {
        int x = read(), y = read(), z = read();
        v[x].push_back(MP(y, z));
        v[y].push_back(MP(x, z));
    }
    Dij();
    for(int i = 1; i <= N; i++) {
        fa[i] = i;
        for(int j = 0; j < v[i].size(); j++) {
            int to = v[i][j].fi, w = v[i][j].se;
            if(top[to] == i || top[i] == to) continue;
            E[++num] = (Edge) {i, to, w};
        }
    }
    sort(E + 1, E + num + 1, comp);
    for(int i = 1; i <= num; i++) {
        solve(E[i].u, E[i].v, dis[E[i].u] + dis[E[i].v] + E[i].w);
        if(cnt == N - 1) break;
    }
    for(int i = 2; i <= N; i++)
        printf("%d\n", ans[i] ? ans[i] : -1);
    return 0;
}