P2121 拆地毯最小生成树

最新推荐文章于 2024-10-12 11:03:43 发布

Rachel caramel

最新推荐文章于 2024-10-12 11:03:43 发布

阅读量96

点赞数

分类专栏：数据结构文章标签： c++ 算法

本文链接：https://blog.csdn.net/Rachelhello123/article/details/123155595

版权

数据结构专栏收录该内容

53 篇文章 4 订阅

订阅专栏

本文介绍了一种利用最大生成树算法求解网络中边权优化的问题。通过实例展示了如何使用C++实现并计算边权和节点连接次数，适用于网络流量分配和最短路径计算。思路类似于 Kruskal 或 Prim 算法，核心在于合并操作和找到关键连接节点。

摘要由CSDN通过智能技术生成

题目：
在这里插入图片描述
输入输出样例
input:
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3
output:
3
-1
3
思路：
类似最大生成树

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>
#include <queue>
#define maxn 111111
using namespace std;
int n,m,k,u,v,w;
int fa[maxn];
struct edge
{
    int x,y,z;
    bool operator<(const edge &e1)const
    {
        return z<e1.z;
    }
}e[maxn];
int getfa(int x)
{
    return (x==fa[x])?x:(fa[x]=getfa(fa[x]));
}
void merge(int x,int y)
{
    int xx=getfa(x),yy=getfa(y);
    if (xx==yy) return;
    fa[xx]=yy;
}
double x[maxn],y[maxn];
int main()
{
    cin>>n>>m>>k;
    int ans=0,cnt=0;
    for (int i=1;i<=n;i++) fa[i]=i;
	for (int i=1;i<=m;i++) 
	{
        cin>>u>>v>>w;
        e[++cnt]={u,v,w};
    }
    sort(e+1,e+cnt+1);
    int cntt=0;
    for(int i=cnt;i>=1;i--)
    {
        if(getfa(e[i].x)!=getfa(e[i].y))
        {
            merge(e[i].x,e[i].y);
            ans+=e[i].z;
            cntt++;
        }
        if(cntt==k) break;
    }
    cout<<ans<<endl;
    return 0;
}