zoukankan      html  css  js  c++  java
  • 牛客 滑雪与时间胶囊(最小生成树)

    这题是一个有向图,因为他必须要从高往低,因此我们在建边时注意一下。这不是普通的最小生成树

    因为有向图的叫最小树形图,但是我们可以用这个思想来做。首先把所有满足条件的边加到结构体中

    其次,先做一下bfs遍历标记所有能到的点,之后把结构体按出边的高度为第一排序,按权值为第二排序

    这么做的原因是,首先我们要保证最多的点,因此肯定所有点都要遍历到,其次,因为都是从高点像低点走。

    所以其实是从1一层一层往外,只有高度高的先走了,高度低的才能走。也就是我们在kruscal时要求a-b这条边,a必须已经在集合中才可以。

    其次这样排序不会影响答案,因为最后所有点的都会在集合中,所以a-b这条边如果在a已经加入集合的情况下还是最小的,他还是会被加入最小生成树。

    排完序后做一遍生成树即可。

    #include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<cmath>
    #include<functional>
    #include<string>
    #include<algorithm>
    #include<iostream>
    #include<set>
    #include<vector>
    #include<queue>
    #include<cstdlib>
    using namespace std;
    typedef long long ll;
    const int N=2e6+10;
    int a[N];
    int h[N],ne[N],e[N],w[N],idx;
    void add(int a,int b,int c){
        e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
    }
    struct node{
        int x,b,c;
        bool operator <(const node &t) const{
            if(a[b]!=a[t.b])
                return a[b]>a[t.b];
            return c<t.c;
        }
    }s[N];
    int n,m;
    int cnt=1;
    int vis[N];
    void bfs(){
        queue<int> q;
        q.push(1);
        vis[1]=1;
        while(q.size()){
            int t=q.front();
            q.pop();
            int i;
            for(i=h[t];i!=-1;i=ne[i]){
                int j=e[i];
                if(!vis[j]){
                    vis[j]=1;
                    q.push(j);
                }
            }
        }
    }
    int p[N];
    int find(int x){
        if(x!=p[x]){
            p[x]=find(p[x]);
        }
        return p[x];
    }
    ll kruscal(){
        sort(s+1,s+1+cnt);
        ll res=0;
        int i;
        for(i=1;i<=n;i++)
            p[i]=i;
        for(i=1;i<cnt;i++){
            int x=s[i].x,b=s[i].b,c=s[i].c;
            if(!vis[x]||!vis[b])
                continue;
            int pa=find(x),pb=find(b);
            if(pa!=pb){
                p[pa]=pb;
                res+=c;
    
            }
        }
        return res;
    }
    int main(){
        int i;
        cin>>n>>m;
        memset(h,-1,sizeof h);
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        for(i=1;i<=m;i++){
            int u,v,c;
            scanf("%d%d%d",&u,&v,&c);
            if(a[u]>=a[v]){
              s[cnt++]=node{u,v,c};
              add(u,v,c);
            }
            if(a[v]>=a[u]){
                add(v,u,c);
                s[cnt++]=node{v,u,c};
            }
        }
        bfs();
        ll res=kruscal();
        int ans=0;
        for(i=1;i<=n;i++){
            if(vis[i])
                ans++;
        }
        cout<<ans<<" "<<res<<endl;
    }
    View Code
  • 相关阅读:
    mac 安装Windows系统
    各种镜像源
    应用官方下载地址汇总
    centos7 升级openssh
    ubuntu16.04升级openssh
    腾讯云
    msdeploy 远程发布到lls
    Java Script 什么是闭包?
    JavaScript我的怀疑
    HTML 之 js是干什么的
  • 原文地址:https://www.cnblogs.com/ctyakwf/p/12930045.html
Copyright © 2011-2022 走看看