BZOJ2753: [SCOI2012]滑雪与时间胶囊【最小生成树】

2753: [SCOI2012]滑雪与时间胶囊

转化模型，就是最小树形图，有向图最小生成树。

看数据范围，好像不能用朱刘算法。

我们发现这是一个DAG，那么我们可以先按照高度排序，然后按照权值排序，这样就能保证是一棵树了。

#include<queue>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN=1e5+5,MAXE=1e6+5;
int n,m,Top,Sum,H[MAXN],fa[MAXN];bool vis[MAXN],Walk[MAXE<<1];long long Ans;
struct xcw{
	int x,y,w;
	bool operator <(const xcw b)const{return H[y]>H[b.y]||(H[y]==H[b.y]&&w<b.w);}
}a[MAXE<<1];
#include<cctype>
int read(){
	int ret=0;char ch=getchar();bool f=1;
	for(;!isdigit(ch);ch=getchar()) f^=!(ch^'-');
	for(; isdigit(ch);ch=getchar()) ret=(ret<<1)+(ret<<3)+ch-48;
	return f?ret:-ret;
}
struct Edge{
	int tot,lnk[MAXN],nxt[MAXE<<1],son[MAXE<<1];
	void Add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
}E;
int get(int x){return fa[x]==x?x:fa[x]=get(fa[x]);}
void Kruskal(){
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=Top;i++)
	if(Walk[i]){
		int x=get(a[i].x),y=get(a[i].y);
		if(x==y) continue;
		fa[y]=x;Ans+=a[i].w;
	}
}
queue<int> que;
void BFS(int x){
	que.push(x);vis[x]=1;
	while(!que.empty()){
		x=que.front();que.pop();Sum++;
		for(int j=E.lnk[x];Walk[j]=1,j;j=E.nxt[j])
		if(!vis[E.son[j]]) que.push(E.son[j]),vis[E.son[j]]=1;
	}
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++) H[i]=read();
	for(int i=1;i<=m;i++){
		int x=read(),y=read(),w=read();
		if(H[x]<H[y]) std::swap(x,y);
		a[++Top]=(xcw){x,y,w};
		if(H[x]==H[y]) a[++Top]=(xcw){y,x,w};
	}
	std::sort(a+1,a+1+Top);
	for(int i=1;i<=Top;i++) E.Add(a[i].x,a[i].y);
	BFS(1);Kruskal();
	printf("%d %lld\n",Sum,Ans);
	return 0;
}