牛客练习赛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[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");elseputs("Yes");
}

转载于:https://www.cnblogs.com/wmj6/p/10661611.html