为什么这道题一定要考虑高度?
查看原帖
为什么这道题一定要考虑高度?
528466
FarmerZhuGe楼主2023/8/25 20:59

这个过不了

#include<cstring>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<vector>
#define x first
#define y second
using namespace std;
typedef long long LL;
typedef pair<int,int> pii;
const int N=1e5+10,M=2e6+10;
int n,m;
int d[N];
int h[N],e[M],ne[M],w[M],idx;
bool st[N];
LL cnt,res;
void add(int a,int b,int c){
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
void Prim(){
	priority_queue<pii,vector<pii>,greater<pii> > q;
	q.push({0,1});
	while(q.size()){
		pii t=q.top();
		q.pop();
		if(st[t.y]) continue;
		st[t.y]=true;
		cnt++,res+=t.x;
		for(int i=h[t.y];~i;i=ne[i]){
			int j=e[i];
			if(st[j]) continue;
			q.push({w[i],j});
		}
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&d[i]);
	memset(h,-1,sizeof h);
	while(m--){
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		if(d[a]>=d[b]) add(a,b,c);
		if(d[b]>=d[a]) add(b,a,c);
	}
	Prim();
	printf("%lld %lld",cnt,res);
	return 0;
}

这个就能过

#include<cstring>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<vector>
using namespace std;
typedef long long LL;
const int N=1e5+10,M=2e6+10;
int n,m;
int d[N];
int h[N],e[M],ne[M],w[M],idx;
bool st[N];
LL dis[N];
LL cnt,res;
struct Node{
	int h,d,id;
	bool operator< (const Node &t)const{
		if(h!=t.h) return h<t.h;
		return d>t.d;
	}
};
void add(int a,int b,int c){
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
void Prim(){
	priority_queue<Node> q;
	memset(dis,0x7f,sizeof dis);
	dis[1]=0;
	q.push({d[1],0,1});
	while(q.size()){
		Node t=q.top();
		q.pop();
		if(st[t.id]) continue;
		st[t.id]=true;
		cnt++,res+=dis[t.id];
		for(int i=h[t.id];~i;i=ne[i]){
			int j=e[i];
			if(st[j]) continue;
			if(dis[j]>w[i]) dis[j]=w[i],q.push({d[j],w[i],j});
		}
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&d[i]);
	memset(h,-1,sizeof h);
	while(m--){
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		if(d[a]>=d[b]) add(a,b,c);
		if(d[b]>=d[a]) add(b,a,c);
	}
	Prim();
	printf("%lld %lld",cnt,res);
	return 0;
}
2023/8/25 20:59
加载中...