牛客练习赛43-C Tachibana Kanade Loves Review

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

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

题目描述

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

说明

立华奏已经学习过了第 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.

备注:

0⩽k⩽n⩽106,m⩽5×106,t⩽1018,Ti,Hi⩽103

地址：https://ac.nowcoder.com/acm/contest/548/C

思路：

一、优先队列，将未选取的点加入优先队列Q,按照价值小的先出队列，对于每个Q.top() 做标记，其价值加入答案，更新与之相连且未选取的点，若更新成功即加入队列，直到所有的点都被选取（遗憾的是TLE....）

二、最小生成树，首先将所有的点与一个虚拟的点a[0]直接相连，边权为a[i]的价值，这样就转换成最小生成树。

Code 1:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
typedef long long LL;
typedef pair<int,int> pr;

const int MAX_N=1e6+5;
const int MAX_M=5e6+5;
int n,m,p;
LL s;
int a[MAX_N];
vector<pr>	e[MAX_N];
bool boo[MAX_N];
priority_queue<pr,vector<pr>,greater<pr> > Q;

int read(){
    int x=0;char s=getchar();
    while(s<'0'||s>'9'){s=getchar();}
    while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
    return x;
}
int main()
{
	n=read();	m=read();	p=read();
	scanf("%Ld",&s);
	for(int i=1;i<=n;++i)
		a[i]=read();
	for(int i=0,x;i<p;++i)
	{
		x=read();
		boo[x]=true;
	}
	int u,v,w;
	for(int i=0;i<m;++i)
	{
		u=read();	v=read();	w=read();
		e[u].push_back({v,w});
		e[v].push_back({u,w});
		if(boo[u])	a[v]=min(a[v],w);
		if(boo[v])	a[u]=min(a[u],w);
	}
	LL Sum=0;
	for(int i=1;i<=n;++i)
		if(!boo[i])	Q.push({a[i],i});
	int vi,id;
	while(!Q.empty()&&p<=n){
		vi=Q.top().first;	id=Q.top().second;
		Q.pop();
		if(boo[id])	continue;
		Sum+=vi;	boo[id]=true;	++p;
		if(Sum>s)	break;
		for(auto c:e[id])
			if(!boo[c.first]&&a[c.first]>c.second){
				a[c.first]=c.second;
				Q.push({a[c.first],c.first});
			}
	}
	if(Sum<=s)	printf("Yes\n");
	else	printf("No\n");
	
	return 0;
}

Code 2：

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
 
const int MAX_N=1e6+5;
const int MAX_M=6e6+5;
struct node{
	int u;
	int v;
	int w;
	bool operator<(const node &A)const{
	  return w<A.w;
	}
};
int n,m,p;
LL s;
int a[MAX_N];
node d[MAX_M];
int id[MAX_N];

int read(){
    int x=0;char s=getchar();
    while(s<'0'||s>'9'){s=getchar();}
    while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
    return x;
}
int Find(int x){
	if(id[x]!=x)	id[x]=Find(id[x]);
	return id[x];
}
int main()
{
	int mm=0;
	n=read();	m=read();	p=read();
	scanf("%Ld",&s);
    for(int i=0;i<=n;++i)
    	id[i]=i;
    for(int i=1;i<=n;++i)
		a[i]=read();
    for(int i=0,x;i<p;++i)
    {
      	x=read();
        a[x]=0;
    }
    for(int i=1;i<=n;++i)
    	d[mm++]={0,i,a[i]};
    int u,v,w;
    for(int i=0;i<m;++i)
    {
    	u=read();	v=read();	w=read();
        d[mm++]={u,v,w};
    }
    sort(d,d+mm); 
    LL Sum=0;
    for(int i=0;i<mm;++i)
    {
    	u=Find(d[i].u);	v=Find(d[i].v);	w=d[i].w;
		if(u!=v){
			id[u]=v;
			Sum+=w;
		}
	}
	if(Sum<=s)   printf("Yes\n");
	else    printf("No\n");
	
    return 0;
}