滑雪与时间胶囊（bzoj 2753）

/*
    单向边的最小生成树
    以末尾点的高度为第一关键字，边长为第二关键字排序，跑最小生成树。 
*/
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define N 100100
#define inf 2000000000
using namespace std;
int h[N],fa[N],dis[N],head[N],to[N*20],next[N*20],n,m,cnt,sum;
long long ans;bool vis[N];
struct node{int u,v,w;}e[N*10];
void add(int u,int v){
    to[++cnt]=v;next[cnt]=head[u];head[u]=cnt;
}
void bfs(){
    queue<int> q;q.push(1);vis[1]=1;
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=head[u];i;i=next[i])
            if(!vis[to[i]]){
                vis[to[i]]=1;
                q.push(to[i]);
            }
    }
}
bool cmp(const node&s1,const node&s2){
    if(h[s1.v]==h[s2.v]) return s1.w<s2.w;
    return h[s1.v]>h[s2.v];
}
int find(int x){
    if(fa[x]==x) return x;
    return fa[x]=find(fa[x]);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&h[i]);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        if(h[e[i].u]<h[e[i].v]) swap(e[i].u,e[i].v);
        add(e[i].u,e[i].v);
        if(h[e[i].v]==h[e[i].u]) add(e[i].v,e[i].u);
    }
    bfs();
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++){
        if(!vis[e[i].u]||!vis[e[i].v]) continue;
        int a=find(e[i].u),b=find(e[i].v);
        if(a!=b) fa[b]=a,sum++,ans+=e[i].w;
    }
    printf("%d ",sum+1);cout<<ans;
    return 0;
}