hdu 1102 (最小生成树)

最新推荐文章于 2018-09-03 09:52:04 发布

原创最新推荐文章于 2018-09-03 09:52:04 发布 · 435 阅读

0 ·

CC 4.0 BY-SA版权

刷题代码专栏收录该内容

75 篇文章

订阅专栏

本文详细介绍了最小生成树的概念及其求解方法——Prim算法，并通过实例展示了如何利用Prim算法解决实际问题，特别是对已建立连接的村庄进行路径优化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

此题也是最小生成树，是Prim算法的直接套用，只需要处理一下，对于村庄中已经修好的路，将两村中间的距离设为0即可。然后直接用Prim算法。

# include <stdio.h>
# include <string.h>
# include <stdlib.h>

# define INF 10000000

int N,Q;
int graph[105][105];
int visit[105],dis[105];

void Init()
{
    int i,j;

    memset(visit,0,sizeof(visit));
    for(i = 1;i <= N;i ++)
        for(j = 1;j <= N;j ++)
            graph[i][j] = graph[j][i] = INF;
    for(i = 1;i <= N;i ++)
        graph[i][i] = 0;
}

int work()
{
    int i,j,sum = 0,mincost,index;

    for(i = 1;i <= N;i ++)
        dis[i] = graph[1][i];
    visit[1] = 1;
    for(i = 1;i < N;i ++)
    {
        mincost = INF;
        for(j = 1;j <= N;j ++)
        {
            if(!visit[j] && dis[j] < mincost)
            {
                mincost = dis[j];
                index = j;
            }
        }

        sum += mincost;
        visit[index] = 1;

        for(j = 1;j <= N;j ++)
        {
            if(!visit[j] && graph[index][j] < dis[j])
                dis[j] = graph[index][j];
        }
    }

    return sum;
}

int main()
{
    int i,j,a,b;

    while(~scanf("%d",&N))
    {
        Init();
        for(i = 1;i <= N;i ++)
            for(j = 1;j <= N;j ++)
                scanf("%d",&graph[i][j]);

        scanf("%d",&Q);
        for(i = 1;i <= Q;i ++)
        {
            scanf("%d%d",&a,&b);
            graph[a][b] = graph[b][a] = 0;
        }
        printf("%d\n",work());
    }

    return 0;
}