这个过不了
#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;
}