链接:https://ac.nowcoder.com/acm/contest/548/C
最小生成树
题目求能不能在t天内学完所有知识点,将每个知识点看成一个顶点,知识点间的联系看成边,这样问题就简化成了一个求最小生成树的问题了。但是还有一些点之间是没有联系的,即图是不连通的,所以我们需要引入一个节点0,将(0,i)这条边的权值定为Ti,这样图就联通了。另外就是条件k了,k个已经学习的知识点表明(0,ki)这条路径已经加入到最小生成树中,只需用并查集合并就好。
#include<bits/stdc++.h>
using namespace std;
const int N=1e6 + 10;
const int M=7e6 + 10;//一共有m+n条边 不知道为什么6e6+10 就会超时
int n,m,k;
long long t,sum=0;
int T;
struct Edge {
int form,to,w;
bool operator < (const Edge x)const{
return w<x.w;
}
}e[M];
int cnt=0;
void addEdge(int u,int v,int w){
e[++cnt].to=v;
e[cnt].w=w;
e[cnt].form=u;
}
int f[N];
int find(int x){
return f[x]==x? x: f[x]=find(f[x]);
}
inline int read(){//快读
long long x=0,f=1;char c=getchar();
while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
return x*f;
}
int main(){
n=read();m=read();k=read();t=read();
//printf("%d %d %d %lld\n",n,m,k,t);
f[0]=0;
for(int i=1;i<=n;i++){
f[i]=i;
T=read();
addEdge(0,i,T);//每个节点和虚拟节点0创建边 边权为点代价
}
int tmp;
for(int i=1;i<=k;i++)
{
tmp=read();
//printf("%d\n",tmp);
f[tmp]=0;//已有节点和虚拟节点0连接
}
int x,y,h;
for(int i=0;i<m;i++){
x=read();y=read();h=read();
//printf("%d %d %d\n",x,y,h);
addEdge(x,y,h);
}
sort(e+1,e+cnt+1);
int ans=k;
for(int i=1;i<=cnt;i++){
x=find(e[i].form),y=find(e[i].to),h=e[i].w;
if(x!=y){
f[x]=y;
sum+=h;
ans++;
}
else continue;
if(ans==n-1||sum>t) break;
}
//cout<<sum;
if(sum>t)printf("No\n");
else printf("Yes\n");
return 0;
}