1 Star 0 Fork 1

saigon/Algorithms

forked from charlieshu/Algorithms 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
P1550 [USACO08OCT]Watering Hole G.cpp 899 Bytes
一键复制 编辑 原始数据 按行查看 历史
charlie 提交于 2024-01-09 00:01 . move from github to gitee
#include <iostream>
#include <algorithm>
using namespace std;
struct edge{
int s,t,len;
};
bool cmp(edge a,edge b){
return a.len<b.len;
}
int find(int *f,int x){
if(f[x] == x) return x;
return f[x] = find(f,f[x]);
}
int main(){
int n,m,edge_sum=0,ans=0;
cin>>n;
n++;
m = n*(n-1)/2;
int f[n];
edge e[m];
for(int i=0;i<n;i++)
f[i] = i;
for(int i=0;i<n-1;i++){
cin>>e[i].len;
e[i].s = n-1;
e[i].t = i;
// e[i].s--;e[i].t--;
}
int k=n-1;
for(int i=0;i<n-1;i++){
for(int j=0;j<n-1;j++){
int len;
cin>>len;
if(j > i){
e[k].s = i;
e[k].t = j;
e[k].len = len;
k++;
}
}
}
sort(e,e+m,cmp);
for(int i=0;i<m;i++){
if(edge_sum == n-1) break;
int f1=find(f,e[i].s),f2=find(f,e[i].t);
if(f1 != f2){
edge_sum++;
ans += e[i].len;
f[f1] = f[f2];
}
}
cout<<ans;
return 0;
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/saigonshu/algorithm.git
git@gitee.com:saigonshu/algorithm.git
saigonshu
algorithm
Algorithms
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385