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

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

题目描述

立华奏是一个刚刚开始学习 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

思路:创建一个虚拟节点(我是用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[13]={0,31,28,31,30,31,30,31,31,30,31,30,31};
int dir[4][2]={1,0 ,0,1 ,-1,0 ,0,-1};
int dirs[8][2]={1,0 ,0,1 ,-1,0 ,0,-1, -1,-1 ,-1,1 ,1,-1 ,1,1};
const int inf=0x3f3f3f3f;
const ll mod=1e9+7;
const int maxn=7e6 + 100;
inline int read(){
    char ch = getchar(); ll x = 0, f = 1;
    while(ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
    while('0' <= ch && ch <= '9') {x = x * 10 + ch - '0'; 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[12000007];
int f[maxn];
int cnt=0;
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=1;i<=n;i++){ //给虚拟节点加上初始边 
        time=read();
        add(0,i,time);
        f[i]=i;
    }
    for(int i=1;i<=k;i++){
        learn=read(); //已经学习的点就和0关联起来就行 
        f[find(learn)]=0;
    }
    for(int i=1;i<=m;i++){ //关联点加边 
        x=read(); y=read(); h=read();
        add(x,y,h);
    }
    sort(edge+1,edge+1+cnt);
    ll ans=0;
    int num=k;
    for(int i=1;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");
}