注解:
1.算法:pirm。
2.存边:链式前向星。
3.找与之关联的最小边:priority_queue。
知识点:
关于priority_queue。
代码注解:
vtct:已经在确定的点的数目;
vt[]:已经确定的点为1,否则为0;
d:该边的长度。
nextv:该边到达的点的编号。
俺的ac代码:
#include<iostream>
#include<queue>
using namespace std;
struct node{
int d,nextv;
bool operator <(const node &n1)const{
if(d==n1.d) return nextv<n1.nextv; //从大到小
else return d>n1.d; //从小到大
}
};
int n,m,sum,vtct;
int vt[5100];
int h[5100],ne[400100],v[400100],w[400100];
int idx;
void add(int x,int y,int z){
ne[++idx]=h[x]; h[x]=idx; v[idx]=y; w[idx]=z;
}
void prim(){
priority_queue<node>pq;
vt[1]=1;
vtct++;
for(int i=h[1];i;i=ne[i])
pq.push({w[i],v[i]});
while(vtct<n&&!pq.empty()){
node tmp=pq.top();
pq.pop();
if(vt[tmp.nextv]) continue;
sum+=tmp.d;
vtct++;
vt[tmp.nextv]=1;
for(int i=h[tmp.nextv];i;i=ne[i])
pq.push({w[i],v[i]});
}
}
int main(){
cin>>n>>m;
if(m<n-1){
printf("orz"); return 0;
}
while(m--){
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z),add(y,x,z);
}
prim();
if(vtct==n) printf("%d",sum);
else printf("orz");
return 0;
}