牛客第69次月赛补题D旅游

最新推荐文章于 2024-10-17 21:05:59 发布

zhensi666

最新推荐文章于 2024-10-17 21:05:59 发布

阅读量69

点赞数 1

文章标签： c++ 算法 c语言数据结构

本文链接：https://blog.csdn.net/zhensi666/article/details/129769684

版权

原题链接：https://ac.nowcoder.com/acm/contest/52441/D

先放一下AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1e5+10;
int f[N];
struct ss{
    int x,y,a;
}s[N];
int n,m;
long long c;
bool cmp(ss x,ss y){
    return x.a<y.a;
}
int find(int x){
    if(x!=f[x])f[x]=find(f[x]);
    return f[x];
}
bool check(int x){
    long long sum=0;
    int u[N]={0};
    for(int i=1;i<=n;i++)f[i]=i;
    int j=0;
    for(int i=1;i<=m;i++){
        int a=s[i].x,b=s[i].y;
        a=find(a),b=find(b);
        if(a!=b){
            f[a]=b;
            if(s[i].a>x){
                u[j++]=s[i].a;
            }
        }
    }
    sort(u,u+j);
    for(int i=j-1;i>=0;i--)sum+=(long long)u[i]*(j-i);
    if(sum>c)return false;
    return true;
}
int main()
{
    cin>>n>>m>>c;
    for(int i=1;i<=m;i++)cin>>s[i].x>>s[i].y>>s[i].a;
    sort(s+1,s+m+1,cmp);
    int l=0,r=1e9;
    while(l<r){
        int mid=(l+r)>>1;
        if(check(mid))r=mid;
        else l=mid+1;
    }
    cout<<l;
    return 0;
}

这道题一开始没有用并查集去做，而是想的开一个st数组存连通的城市，结果只能过50%。

最后上网查了一下佬们的思路发现并查集能过。

于是在这里填一个坑

之前的错误代码粘这里了，希望佬们帮看看

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1e5+10;
struct ss{
    int x,y,a;
}s[N];
int n,m;
long long c;
bool cmp(ss x,ss y){
    return x.a<y.a;
}
bool check(int x){
    long long sum=0;
    int u[N]={0};
    bool st[N]={0};
    int j=0;
    for(int i=1;i<=m;i++){
        if(!st[s[i].x]||!st[s[i].y]){
            if(s[i].a<=x){
                st[s[i].x]=1,st[s[i].y]=1;
            }
            else{
                u[j++]=s[i].a;
                st[s[i].x]=1,st[s[i].y]=1;
            }
        }
    }
    sort(u,u+j);
    for(int i=j-1;i>=0;i--)sum+=(long long)u[i]*(j-i);
    if(sum>c)return false;
    for(int i=1;i<=n;i++)if(!st[i])return false;
    return true;
}
int main()
{
    cin>>n>>m>>c;
    for(int i=1;i<=m;i++)cin>>s[i].x>>s[i].y>>s[i].a;
    sort(s+1,s+m+1,cmp);
    int l=0,r=1e9;
    while(l<r){
        int mid=(l+r)>>1;
        if(check(mid))r=mid;
        else l=mid+1;
    }
    cout<<l;
    return 0;
}