LightOJ 1040 – Donation 【Prim】

最新推荐文章于 2024-11-13 00:06:02 发布

mfcheer

最新推荐文章于 2024-11-13 00:06:02 发布

阅读量779

点赞数

CC 4.0 BY-SA版权

分类专栏： ----------生成树

本文链接：https://blog.youkuaiyun.com/u014427196/article/details/46380521

----------生成树专栏收录该内容

28 篇文章

订阅专栏

本文提供了一种解决LightOJ 1040问题的算法实现，采用Prim算法寻找最小生成树，并通过示例代码展示了如何初始化矩阵和进行最小生成树计算的过程。

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

题目连接：http://www.lightoj.com/volume_showproblem.php?problem=1040

题意也不说了。

水题

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <string>
#include <algorithm>
#include <math.h>
#include <ctype.h>
#include <time.h>
#include <queue>
#include <iterator>
#include <vector>
#include <set>

using namespace std;

const int MAXN = 55;
const int inf = 100000000;

int p[MAXN][MAXN];
int n;

int dis[MAXN], v[MAXN];
int prim(int map[][55], int n)
{
    memset(v, 0, sizeof(v));
    int min, sum = 0;
    for (int i = 0; i < n; i++)
        dis[i] = map[0][i];
    v[0] = 1;
    for (int i = 1; i < n; i++)
    {
        int k = -1;
        min = inf;
        for (int j = 0; j < n; j++)
            if (!v[j] && min>dis[j])
            {
                k = j;
                min = dis[j];
            }
        if (k == -1)
            return - 1;
        sum += min;
        v[k] = 1;
        for (int j = 0; j < n; j++)
            if (!v[j] && dis[j]>map[k][j])
                dis[j] = map[k][j];
    }
    return sum;
}

void init()
{
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            if (i != j && p[i][j] == 0 && p[j][i] == 0)
            {
                p[i][j] = p[j][i] = inf;
            }
        }
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    for (int cases = 1; cases <= t; cases++)
    {
        int ans = 0;
        memset(p,0,sizeof(p));
        scanf("%d",&n);
        for (int i = 0; i < n; i++)
            for (int j = 0; j < n; j++)
            {
                scanf("%d",&p[i][j]);
                ans += p[i][j];
                if (i == j)
                    p[i][i] = 0;
                if (i!=j && p[i][j] == 0)
                {
                    p[i][j] = p[j][i];
                }
                else if (i != j && p[j][i] != 0 && p[i][j]!=0)
                {
                    p[i][j] = p[j][i] = min(p[i][j], p[j][i]);
                }
                else if (i != j && p[j][i] == 0)
                {
                    p[j][i] = p[i][j];
                }
            }
        init();
        int tmp = prim(p, n);
        ans -= tmp;
        if (tmp!=-1)
            printf("Case %d: %d\n",cases,ans);
        else
            printf("Case %d: -1\n", cases);
    }
    return 0;
}