题目
题目详解
- 就一非常简单的最小生成树问题而已,但是就是耗了我很长时间也不知错哪了(现在也不清楚)。。
我用的Kruskal算法实现最小生成树,而我是用的最小堆实现的,然后最小堆实现就硬是给我卡最后一个用例,我也不知道为什么?然后改普通数组排序就直接过了。。。
以下为最小堆实现的main函数,然后就硬是卡最后一个用例,把最小堆换数组存值后就过了。。
int main() {
int c,n,m;
int u,v,w;
UnionFind uf(101);
while (scanf("%d%d%d",&c,&n,&m)!=EOF){
priority_queue<node>pq;
while (n--){
scanf("%d %d %d",&u,&v,&w);
pq.push({u,v,w});
}
int cnt = m-1;
int val = 0;
while (!pq.empty()&&cnt>0){
auto & t = pq.top();pq.pop();
if(!uf.isconnected(t.u,t.v)){
uf.merge(t.u,t.v);
cnt--; val+=t.w;
}
}
const char* str;
if(cnt==0&&val<=c) str = "Yes";
else str = "No";
printf("%s\n",str);
uf.clear();
}
return 0;
}
经验教训:竞赛OJ的存图题目可能有时候比较玄乎,比如我之前写的一道图论题目,是最短路的那种,然后他两个结点直接竟然存在多条边?我™。。。然后我就不得不用二维矩阵存图,而这次我猜又是这种类似的输入骚操作,导致我只能用数组存边。。
所以总结就是这种竞赛题,写最小生成树,最好使用数组存边!
解题代码
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
//TODO 并查集实现
class UnionFind{
private:
int* root;
int* rank;
int length;
public:
UnionFind(int size):length(size){
root = new int[size];
rank = new int[size];
for(int i=0;i<length;i++){
root[i] = i;
rank[i] = 1;
}
}
int find(int x){
if(root[x]==x)
return x;
return root[x]=find(root[x]);
}
void merge(int x,int y){
int rootX = find(x);
int rootY = find(y);
//高度小的树被高度大的合并,如果高度一致合并后高度增加
if(rootX!=rootY){
if(rank[rootX]>rank[rootY]){
root[rootY] = rootX;
} else if(rank[rootX]<rank[rootY]){
root[rootX] = rootY;
}else{
root[rootY] = rootX;
rank[rootX]++;
}
}
}
bool isconnected(int x,int y){
return find(x)==find(y);
}
void clear(){
for(int i=0;i<length;i++){
root[i] = i;
rank[i] = 1;
}
}
};
//TODO 定义用于存边的结构
struct node{
int u,v,w;
bool operator<(const node& a)const {
return this->w<a.w;
}
}edges[10005];
int main() {
int c,n,m;
int u,v,w;
UnionFind uf(105);
while (scanf("%d%d%d",&c,&n,&m)!=EOF){
for(int i=0;i<n;i++){
scanf("%d%d%d",&u,&v,&w);
edges[i] = {u,v,w};
}
int cnt = m-1;
int val = 0;
sort(edges,edges+n);
for (int i=0;i<n&&cnt>0;i++){
node t = edges[i];
if(!uf.isconnected(t.u,t.v)){
uf.merge(t.u,t.v);
cnt--; val+=t.w;
}
}
const char* str;
if(cnt==0&&val<=c) str = "Yes";
else str = "No";
printf("%s\n",str);
uf.clear();
}
return 0;
}