求助大佬,这个怎么写错了?
查看原帖
求助大佬,这个怎么写错了?
141577
yhc12345楼主2020/8/4 13:34
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#define INF 0x3f3f3f3f

using namespace std;
typedef long long ll;
int n,m;
const int N=5010,M=2e5+10;
struct Edge {
	int v,w,nxt;
}e[M<<1];
int head[N],ne;
int lc[N];
bool vis[N];
void init() {
	memset(head,-1,sizeof(head));
	ne=0;
}
void add_edge(int u,int v,int w) {
	e[ne].v=v;
	e[ne].w=w;
	e[ne].nxt=head[u];
	head[u]=ne++;
}
struct Node {
	int id,lc;
	Node() {
		
	}
	Node(int _id,int _lc) {
		id=_id;
		lc=_lc;
	}
};
struct cmp {
	bool operator() (Node a,Node b) {
		return a.lc>b.lc;
	}
};
priority_queue<Node,vector<Node>,cmp> pq;
bool flag=true;
int ans;
void prim() {
	memset(lc,0x3f,sizeof(lc));
	lc[1]=0;
	vis[1]=true;
	for(int i=head[1];i!=-1;i=e[i].nxt) {
		lc[e[i].v]=e[i].w;
		pq.push(Node(e[i].v,e[i].w));
	}
	for(int i=1;i<n;i++) {
		int k=-1;
		while(!pq.empty()&&(k==-1||vis[k])) {
			k=pq.top().id;
			pq.pop();
		}
		if(k==-1) {
			flag=false;
			break;
		}
		vis[k]=true;
		ans+=lc[k];
		for(int i=head[k];i!=-1;i=e[i].nxt) {
			if(!vis[e[i].v]&&lc[e[i].v]>lc[k]+e[i].w) {
				lc[e[i].v]=lc[k]+e[i].w;
				pq.push(Node(e[i].v,lc[e[i].v]));
			}
		}
	}
}
int main() {
	ios::sync_with_stdio(false);
	init();
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++) {
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add_edge(u,v,w);
		add_edge(v,u,w);
	}
	prim();
	if(!flag) {
		printf("orz");
	}
	else {
		printf("%d",ans);
		
	}
	return 0;
}
2020/8/4 13:34
加载中...