牛客网 牛客练习赛43 C.Tachibana Kanade Loves Review-最小生成树(并查集+Kruskal)+建虚点+读入挂

链接：https://ac.nowcoder.com/acm/contest/548/C
来源：牛客网

Tachibana Kanade Loves Review

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 524288K，其他语言1048576K
64bit IO Format: %lld

题目描述

立华奏是一个刚刚开始学习 OI 的萌新。
最近，实力强大的 $\text{QingyuQingyu 当选了 IODS 9102 的出题人。众所周知， IODS 是一场极其毒瘤的比赛。为了在这次比赛中取得好的成绩，立华奏决定学习可能考到的每一个知识点。 在 QingyuQingyu 的博客中，立华奏得知这场比赛总共会考察选手 n 个知识点。此前，立华奏已经依靠自学学习了其中 k 个知识点。接下来，立华奏需要学习其他的知识点，每学习一个单独的知识点，需要消耗的时间为 TiTi 天。同时，某些知识点之间存在联系，可以加速学习的过程。经过计算，立华奏一共发现了其中 m 种联系，第 i 种联系可以表示为(Xi,Yi,Hi)(Xi,Yi,Hi)，其含义为“在掌握了第 XiXi 个知识点和第 YiYi 个知识点中任意一个后，学习 HiHi 天即可掌握另一个知识点”。 留给立华奏的时间所剩无几，只有 t 天，因此，她想知道自己能不能在这 t 天内学习完成所有的知识点。}$

输入描述:

本题输入量较大，请注意使用效率较高的读入方式
输入的第一行包含四个整数 n, m, k, t，含义见上所述。

接下来一行，包含 n 个整数，依次表示 ${\mathrm{T1,T2,\cdots ,TnT1,T2,\cdots ,Tn}}^{}$

接下来一行，包含 k 个整数，表示立华奏已经学习过的知识点。如果 k=0，则此处为一空行。
接下来 m 行，每行 3 个整数 ${\mathrm{Xi,Yi,HiXi,Yi,Hi，描述一种联系。}}^{}$

输出描述:

如果立华奏能够学习完所有的知识点，输出一行 Yes。否则输出 No

示例1

输入

复制

4 3 2 5
4 5 6 7
2 3
1 2 3
1 3 2
3 4 2

输出

复制

Yes

说明

立华奏已经学习过了第 2, 3 个知识，由第 2 个关系，立华奏可以花 2 天学会知识点 1，在由关系 3， 立华奏可以 2 天学会知识点 4，因此总共需要花费 4 天，可以完成任务。

示例2

输入

复制

5 4 0 12
4 5 6 7 1

1 2 3
1 3 2
3 4 2
1 5 233

输出

复制

Yes

说明

立华奏比较菜，因此什么都没有学过。她可以选择先花 4 天的时间学会知识点 1。然后根据关系 1, 2，分别花 3, 2 天的时间学会知识点 2, 3，再根据关系 3，花 2 天的时间学会知识点 4。然后，她再单独学习知识点 5，花费1天，总共花费了 12 天 ，可以完成任务。

请注意，虽然关系 4 允许立华奏在知识点 1 的基础上学习知识点 5，但需要的时间比单独学习还要多，因此立华奏不会在知识点 1 的基础上学习知识点 5.

备注:

$\mathrm{0⩽k⩽n⩽106,m⩽5\times 106,t⩽1018,Ti,Hi⩽103}$

感觉这道题完全就是面向数据编程，而且数据还有点弱，好多人贪心暴力写的，竟然过了，跑的贼快，也是醉了。

不用读入挂过不了。然后就是我结构体里面写排序竟然比写外面慢？？？！写里面超时了，写外面bool型的过了，队友写里面过了，外面没过，试了三种排序，有点意思，反正乱七八糟的。

建一个虚拟的节点，所有节点都和这个节点相连，然后跑最小生成树。因为有的节点是靠其他节点，有的是直接取自己的值。kruskal直接写主函数里面了。

读入挂，写n=read()的比read(n)的快。

代码:

//C-最小生成树-要用读入挂，然后就是排序函数写外面，写结构体里面也超时了。。。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=6e6+10;

namespace IO{
    char buf[1<<15],*S,*T;
    inline char gc(){
        if (S==T){
            T=(S=buf)+fread(buf,1,1<<15,stdin);
            if (S==T)return EOF;
        }
        return *S++;
    }
    inline int read(){
        int x; bool f; char c;
        for(f=0;(c=gc())<'0'||c>'9';f=c=='-');
        for(x=c^'0';(c=gc())>='0'&&c<='9';x=(x<<3)+(x<<1)+(c^'0'));
        return f?-x:x;
    }
    inline long long readll(){
        long long x;bool f;char c;
        for(f=0;(c=gc())<'0'||c>'9';f=c=='-');
        for(x=c^'0';(c=gc())>='0'&&c<='9';x=(x<<3)+(x<<1)+(c^'0'));
        return f?-x:x;
    }
}
using IO::read;
using IO::readll;

struct edge{
    int u,v;
    ll w;

//    friend bool operator<(edge a,edge b){
//        return a.w<b.w;
//    }

}path[maxn];

bool cmp(edge a,edge b)
{
    return a.w<b.w;
}

int cnt=0;
int fa[maxn];
ll cost[maxn];

int Find(int x)
{
    if(fa[x]==x) return x;
    return fa[x]=Find(fa[x]);
}

int main()
{
    int n,m,k;
    ll t;
    n=read();m=read();k=read();t=readll();
    for(int i=1;i<=n;i++){
        cost[i]=read();
    }
    if(k!=0){
        for(int i=1;i<=k;i++){
            int p;
            p=read();
            cost[p]=0;
        }
    }
    for(int i=1;i<=n;i++){
        path[++cnt]=edge{0,i,cost[i]};
    }
    for(int i=1;i<=m;i++){
        int u,v,w;
        u=read();v=read();w=read();
        path[++cnt]=edge{u,v,w};
    }
//    sort(path+1,path+1+n,[](edge a,edge b){return a.w<b.w;});
    sort(path+1,path+1+n,cmp);
    for(int i=1;i<=n;i++){
        fa[i]=i;
    }
    ll ans=0;
    for(int i=1;i<=cnt;i++){
        int fx=Find(path[i].u);
        int fy=Find(path[i].v);
        if(fx!=fy){
            fa[fx]=fa[fy];
            ans+=path[i].w;
        }
    }
    if(ans<=t) printf("Yes
");
    else printf("No
");
    return 0;
}

。。。