Jzoj P4261 最小代价___spfa+最小生成树

题目大意：

给出一幅由$n$个点$m$条边构成的无向带权图，有些点是黑点，其他点是白点。
现在每个白点都要与他距离最近的黑点通过最短路连接（如果有很多个黑点，可以选取其中任意一个），我们想要使得花费的代价最小。请问这个最小代价是多少？
注意：最后选出的边保证每个白点到离它最近的黑点的距离仍然等于原图中的最短距离。

$1≤n≤100000，1≤m≤200000$
$1≤边权z≤1000000000$

分析：

我们设一个点$S$，连向所有的黑点，
即$S->黑点$，边权为$0$，
然后对给出的边建双向边，
从$S$点开始跑$spfa$，
那么显然
$S$到所有的白点的距离$dis[]$，就是离它最近的黑点的距离，而且这条路上只可能有$1$个黑点，
因为$S$连向所有黑点，而黑点间的边不存在负数，
又因为$S$到所有黑点的边的权值都是$0$，可以认为就是黑点连向白点，
然后将所有满足$dis[x]+cost=dis[y]$的边存起来，$cost$为边权，这样我们其实就是把所有的最短路的边的存起来了，包括到一个点的多条最短路的边。
然后我们通过这些边建一颗最小生成树，就可以保证个白点到离它最近的黑点的距离仍然等于原图中的最短距离，且代价最小。
无法构建最小生成树，则无解

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define N 100005

using namespace std;

typedef long long ll;

queue <int> Q;

struct Node { int To, nxt; ll w; }e[5*N];
struct Text{ int u, v; ll w; }c[5*N];

int ls[N], fa[N], n, m, cnt, tot, num, S = 0;
ll dis[N], ans;
bool vis[N];

void Addedge(int u, int v, ll w) {
    e[++cnt].To = v, e[cnt].w = w, e[cnt].nxt = ls[u], ls[u] = cnt;
}

void spfa() {
    memset(dis, 63, sizeof(dis));
    dis[S] = 0, vis[S] = 1;
    Q.push(S);
    while(!Q.empty()) {
          int u = Q.front();
          Q.pop();
          for (int i = ls[u]; i; i = e[i].nxt) {
               int v = e[i].To;
               if (dis[v] > dis[u] + e[i].w) {
                   dis[v] = dis[u] + e[i].w;
                   if (!vis[v]) 
                       Q.push(v), vis[v] = 1;
               }
          }
          vis[u] = 0;
    }
}


void dfs(int u) {
    for (int i = ls[u]; i; i = e[i].nxt) {
         int v = e[i].To;
         if (dis[u] + e[i].w == dis[v]) {
             dfs(v);
             c[++tot].u = u, c[tot].v = v, c[tot].w = e[i].w;
         }
    }
}

bool cmp(Text aa, Text bb) {
    return aa.w < bb.w;
}

int Find(int  x) {
    if (fa[x] == x) return x;
    fa[x] = Find(fa[x]);
    return fa[x];
}

void Work() {
    sort(c + 1, c + tot + 1, cmp);
    num = 0;
    for (int i = 1; i <= tot; i++) {
         int x = Find(c[i].u);
         int y = Find(c[i].v);
         if (x != y) 
            ++num, ans += c[i].w, fa[y] = x;
         if (num == n) break;
    }
}

int main() {
    scanf("%d %d", &n, &m);
    int u, v; ll w;
    for (int i = 1; i <= n; i++) {
         scanf("%d", &u);
         if (u) Addedge(S, i, 0);
         fa[i] = i;
    }

    for (int i = 1; i <= m; i++) {
         scanf("%d %d %lld", &u, &v, &w);
         Addedge(u, v, w); Addedge(v, u, w);
    }
    spfa();
    dfs(S);
    Work();
    if (num < n) printf("impossible\n");
            else printf("%lld\n", ans);
    return 0;
}