poj 1251 Jungle Roads（最小生成树）

来源：互联网发布：阿里云os应用中心编辑：程序博客网时间：2024/06/03 12:55

一看就是最小生成树

图论很久没看了，想抽时间系统地看下

所以直接把以前的模板贴上了

题目有陷阱就是读入字符如果用scanf来做会很恶心

因为每一行末尾可能有多个空格，只用getchar()会RE

所以直接用cin处理即可

代码如下：

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#define MAXN 1010#define INF 0x7fffffff#define ll long longusing namespace std;int map[MAXN][MAXN];int dis[MAXN];int n;int prim(void){    int now;    int minedge,minnode;                       for(int i=0;i<n;i++)        dis[i] = INF;    now = 0;    int ans = 0;    for(int i=0; i<n-1; i++){        dis[now] = -1;        minedge = INF;        for(int j=0; j<n; j++) {            if(j!=now && dis[j]>=0) {                if(map[now][j] > 0) {                    dis[j] = min(dis[j], map[now][j]);                }                if(dis[j] < minedge) {                    minedge = dis[j];                    minnode = j;                       }                    }                   }        now = minnode;        ans += minedge;                         }    return ans;}int main(void) {    int t, x, y, dis;    char ch;    while(cin >> n, n) {        memset(map, 0, sizeof(map));        for(int c=0; c<n-1; ++c) {            cin >> ch;            x = ch-'A';            cin >> t;             for(int i=0; i<t; ++i) {                cin >> ch >> dis;                y = ch-'A';                map[x][y] = dis;                map[y][x] = dis;            }        }    /*        for(int i=0; i<n; ++i) {            for(int j=0; j<n; ++j) {                if(map[i][j]) {                    printf("map[%c][%c] = %d\t", i+'A', j+'A', map[i][j]);                }            }            printf("\n");        }    */        printf("%d\n", prim());    }    return 0;}

刚刚在练习克鲁斯卡尔算法，对着模板写了下，把代码也贴在这吧

#include <cstdio>#include <iostream>#include <algorithm>#define MAXN 1010#define ll long longusing namespace std;int u[MAXN], v[MAXN], w[MAXN];int p[MAXN], r[MAXN];int n, m, cnt;int cmp(const int i, const int j) {    return w[i]<w[j];}int find(int x) {    return p[x]==x ? x : p[x]=find(p[x]);}int Kruskal() {    int ans = 0;    for(int i=0; i<n; ++i)        p[i] = i;    for(int i=0; i<cnt; ++i)        r[i] = i;    sort(r, r+cnt, cmp);    for(int i=0; i<cnt; ++i) {        int e = r[i];        int x = find(u[e]);        int y = find(v[e]);        if(x != y) {            ans += w[e];            p[x] = y;        }    }    return ans;}int main(void) {    char ch1, ch2;    while(scanf("%d", &n), n) {        cnt = 0;        for(int i=0; i<n-1; ++i) {            getchar();            ch1 = getchar();            scanf("%d", &m);            while(m--) {                getchar();                ch2 = getchar();                u[cnt] = ch1-'A';                v[cnt] = ch2-'A';                scanf("%d", &w[cnt]);                cnt++;            }        }        cout << Kruskal() << endl;    }    return 0;}

0 0