[MST_prim] PKU 1258 AgriNet

prim入门练习题。

prim的流程如下：

// 清空顶点集；

任选一个点u作为起点，加入顶点集，从u出发选一条最小边(u, v)，将v加入顶点集，重复直到所有点都进入顶点集。

其中最小边可以用堆来做，还可以用邻接表加速（这里嫌麻烦）。

# include <cstdio>
# include <queue>

using namespace std;

typedef pair<int, int> pii;

# define N (100 + 5)
# define INF 0x7FFFFFFF

int n;
int g[N][N];

int prim(void)
{
    priority_queue <pii, vector<pii>, greater<pii> > Q;
    int key[N];
    char vis[N];
    for (int i = 1; i <= n; ++i)
    {
        vis[i] = 0;
        key[i] = (i==1 ? 0:INF);
    }
    Q.push(make_pair(key[1], 1));
    int ans = 0;
    while (!Q.empty())
    {
        int u = Q.top().second; Q.pop();
        if (!vis[u]) ans += key[u];
        vis[u] = 1;
        for (int i = 1; i <= n; ++i) if (!vis[i] && g[u][i] < key[i])
        {
            key[i] = g[u][i];
            Q.push(make_pair(key[i], i));
        }
    }
    return ans;
}

void read_graph(void)
{
    for (int i = 1; i <= n; ++i)
    for (int j = 1; j <= n; ++j)
        g[i][j] = INF;
    
    for (int i = 1; i <= n; ++i)
    for (int j = 1; j <= n; ++j)
    {
        int x;
        scanf("%d", &x);
        if (g[i][j] > x) g[j][i] = g[i][j] = x;
    }
}

int main()
{
    while (~scanf("%d", &n))
    {
        read_graph();
        printf("%d\n", prim());
    }
    
    return 0;
}

kruskal相对较简洁，适用于稀疏图。