还是简单的的MST(最小生成树)
数据大意:
N,M 表示图有N个顶点,M条边。
下面有M组数据,a,b,c
依次表示顶点a到b的权值是c
这个题要求的并不是MST,而是要生成树的边的权值总和最大,其实就是MST的变形
不同是多了一个条件,若图中不存在MST,则输出-1,这里没有看到,wrong了一次。
因为题目直接给出的是边的信息,所以这里就用了Kruscal算法
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>
using namespace std;
const int vMax = 1010;
struct Edge
{
int from,to,w;
Edge(int a,int b,int c)
{
from = a;
to = b;
w = c;
}
friend bool operator < (const Edge& a,const Edge& b)
{
return a.w > b.w;
}
};
vector<Edge>v;
int N,M;
int root[vMax];
void init_set(int n)
{
for(int i = 1 ; i <= n ; i++)
{
root[i] = i;
}
}
int find_set(int a)
{
if(root[a] != a)
{
root[a] = find_set(root[a]);
}
return root[a];
}
void union_set(int a,int b)
{
int x = find_set(a);
int y = find_set(b);
if(x != y)
{
root[x] = y;
}
}
void MST()
{
int ans = 0;
for(int i = 0 ; i < v.size() ; i++)
{
int a = v[i].from;
int b = v[i].to;
int c = v[i].w;
if(find_set(a) != find_set(b))//可以合并,合并到同一子树中
{
ans += c;//总权值不断++
union_set(a,b);
}
}
int k = find_set(1);//判断图是否连通
bool flag = false;
for(int i = 2 ; i <= N ; i++)
{
if(find_set(i) != k)
{
flag = true;
}
}
if(flag)
{
ans = -1;
}
printf("%d\n",ans);
}
int main()
{
#ifdef LOCAL
freopen("in.txt","r",stdin);
#endif // LOCAL
while(scanf("%d %d",&N,&M) != EOF)
{
init_set(N);
int a,b,c;
v.clear();
for(int i = 0 ; i < M ; i++)
{
scanf("%d %d %d",&a,&b,&c);
v.push_back(Edge(a,b,c));
}
sort(v.begin(),v.end());//边的权值按照降序排序,这样求出来的生成树的权值是最大的
MST();
}
return 0;
}