继续畅通工程

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 14424    Accepted Submission(s): 6284

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

当N为0时输入结束。

Output

每个测试用例的输出占一行，输出全省畅通需要的最低成本。

Sample Input

3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0

Sample Output

3
1
0

分析：这道题加了点料，区分已建和没建的，所以我们将已建的成本定义为0，就可以了

代码1：（prim）

#include <stdio.h>
#include <string.h>
#define M 105
#define INF 0x3f3f3f3f

int map[M][M], low[M], n;
bool vis[M];

int prim(){
    int i, min, pos = 1, j;
    for(i = 1; i <= n; i ++){
        low[i] = map[pos][i];
        vis[i] = 0;
    }
    int ans = 0;
    vis[pos] = 1;
    for(i = 1; i < n; i++){
        min=  INF;
        for(j = 1; j <= n; j ++){
            if(!vis[j]&&low[j] < min){
                min = low[j]; pos = j;
            }
        }
        if(min == INF) break;
        vis[pos] = 1;
        ans += min;
        for(j = 0; j <= n; j ++)
            if(!vis[j]&&map[pos][j] < low[j])
                low[j] = map[pos][j];
    }
    return ans;
}

int main(){
    while(scanf("%d", &n), n){
        int i, a, b, c, d, j;
        for(i = 1; i <= n; i ++)
        for(j = 1; j <= n; j ++)
        map[i][j] = INF;
        for(i = 0; i < n*(n-1)/2; i++){
            scanf("%d%d%d%d", &a, &b, &c, &d);
            if(d){
                map[a][b] = map[b][a] = 0; continue;
            }
            if(map[a][b] > c)
                map[a][b] = map[b][a] = c;
        }
        printf("%d\n", prim());
    }
    return 0;
}

代码2：（kruskal）

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define M 105
#define INF 0x3f3f3f3f
using namespace std;

struct node{
    int from, to, val;
}s[M*M];
int fat[M], tot, n;

int f(int x){
    if(fat[x] != x) fat[x] = f(fat[x]);
    else return x;
}

int cmp(node a, node b){
    return a.val<b.val;
}

int kruskal(){
    int i;
    for(i = 1; i <= n; i ++) fat[i] = i;
    int ans = 0;
    sort(s, s+tot, cmp);
    for(i = 0; i< tot; i ++){
        //printf("%d%d%d..\n", s[i].from, s[i].to, s[i].val);
        int x = f(s[i].from);
        int y = f(s[i].to);
        //printf("%d..%d..\n", x, y);
        if(x != y){
            ans += s[i].val;
            fat[y] = x;
        }
    }
    return ans;
}

int main(){
    while(scanf("%d", &n), n){
        tot = n*(n-1)/2;
        int i, a, b, c, d;
        for(i = 0; i < tot; i++){
            scanf("%d%d%d%d", &s[i].from, &s[i].to, &s[i].val, &d);
            if(d) s[i].val = 0;
        }
        printf("%d\n", kruskal());

    }
    return 0;
}