4261. 【NOIP2015模拟10.22】最小代价 (Standard IO)

Description

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

Input

第一行两个整数n，m；
第二行n 个整数，0表示白点，1 表示黑点；
接下来m 行，每行三个整数x，y，z，表示一条连接x和y 点，权值为z 的边。

Output

如果无解，输出impossible；
否则，输出最小代价。

Sample Input

5 7
0 1 0 1 0
1 2 11
1 3 1
1 5 17
2 3 1
3 5 18
4 5 3
2 4 5

Sample Output

5
【样例解释】
选 2、4、6三条边

Data Constraint

对30%的输入数据： 1≤n≤10， 1≤m≤20；
对100%的输入数据：1≤n≤100000，1≤m≤200000，1≤z≤1000000000

思路

开个S超级点，和每个黑点的距离为0，然后以S为起点跑spfa，得到一组最短路的数据，之后再用dfs找到每条边，套到最小生成树中，用并查集优化，就可得出答案。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define M 100005
#define inf 0x3f3f3f3f
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define Fill(x) memset(x, inf, sizeof(x))
using namespace std;

typedef long long ll;

queue <int> Q;

struct Node { int To, nxt; ll w; }e[5*M];
struct Text{ int u, uu; ll ss; }p[5*M];

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

void add(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);
             p[++tot].u = u, p[tot].uu = v, p[tot].ss = e[i].w;
         }
    }
}

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

int find(int  x) {
    if (re[x]==x) return x;
    re[x]=find(re[x]);
    return re[x];
}

void work() {
    sort(p+1,p+tot+1,cmp);
    num = 0;
    fo(i,1,tot) {
         int x=find(p[i].u);
         int y=find(p[i].uu);
         if (x != y) 
            ++num, ans += p[i].ss, re[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) add(S, i, 0);
         re[i] = i;
    }

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