牛客练习赛43 Tachibana Kanade Loves Review C(最小生成树Kruskal)

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

题目描述

立华奏是一个刚刚开始学习 OI 的萌新。
最近，实力强大的 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 个整数，依次表示 T1,T2,⋯,TnT1,T2,⋯,Tn

接下来一行，包含 k 个整数，表示立华奏已经学习过的知识点。如果 k=0，则此处为一空行。
接下来 m 行，每行 3 个整数 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

思路:创建一个虚拟节点(我是用0) 然后对于已学的点和没学 以及有关联的点进行建边 最后Kruska跑一边

但是这题卡常 所以写法上要尽量优化

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#define ll long long int
using namespace std;
inline ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
inline ll lcm(ll a,ll b){return a/gcd(a,b)*b;}
int moth[]={,,,,,,,,,,,,};
int dir[][]={, ,, ,-, ,,-};
int dirs[][]={, ,, ,-, ,,-, -,- ,-, ,,- ,,};
const int inf=0x3f3f3f3f;
const ll mod=1e9+;
const int maxn=7e6 + ;
inline int read(){
    char ch = getchar(); ll x = , f = ;
    while(ch < '' || ch > '') {if(ch == '-') f = -; ch = getchar();}
    while('' <= ch && ch <= '') {x = x *  + ch - ''; ch = getchar();}
    return x * f;
}
struct node{
    int to;
    int from;
    int v;
    friend bool operator < (node a,node b){
        return a.v<b.v;
    }
};
node edge[];
int f[maxn];
int cnt=;
int n,m,k;
ll t;
void add(int from,int to,int va){ //加边
    edge[++cnt].to=to;
    edge[cnt].v=va;
    edge[cnt].from=from;
}
int find(int x){
    if(x!=f[x])
       f[x]=find(f[x]);
    return f[x];
}
int main(){
    //ios::sync_with_stdio(false);
    n=read(); m=read(); k=read();
    t=read();
    int time;
    int learn;
    int x,y,h;
    for(int i=;i<=n;i++){ //给虚拟节点加上初始边
        time=read();
        add(,i,time);
        f[i]=i;
    }
    for(int i=;i<=k;i++){
        learn=read(); //已经学习的点就和0关联起来就行
        f[find(learn)]=;
    }
    for(int i=;i<=m;i++){ //关联点加边
        x=read(); y=read(); h=read();
        add(x,y,h);
    }
    sort(edge+,edge++cnt);
    ll ans=;
    int num=k;
    for(int i=;i<=cnt;i++){
        if(find(edge[i].from)!=find(edge[i].to)){
        //    cout<<edge[i].from<<" "<<edge[i].to<<endl;
            f[find(edge[i].from)]=find(edge[i].to);
            ++num;
            ans+=edge[i].v;
        }
        if(num==n||ans>t){
            break;
        }
    }
    if(ans>t)
        puts("No");
    else
        puts("Yes");
}