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

题目
题解

Solution

因为只能从高处到低处，所以无向边可以当有向边看待，然后按照题目意思就是给你一个有向图，求一个最小树形图，然后如果你用朱刘算法来算，就只能得到70分。
这道题具有与其余最小树形图不一样的地方：点有高度！难道高度只是拿来转化为有向边吗？当然不是。 回想kruskal为什么不能求最小树形图？因为每次找的最小边是有向的，所以算法完成之后不能保证根可以到儿子，有可能有反向边！
但是这道题的反向边只会在高度相同的点之间出现。如果把边先按终点高度排序为第一关键字，边长为第二关键字排序之后，就会保证优先到高点，同高点之间选小边，就不会出现反向的情况，所以可以用kruskal

Code

#include<bits/stdc++.h>
using namespace std;
const int M=2e6+1,N=1e5+1;
struct node{
	int x,y,w;
}e[M],c[M];
struct kk{
	int to,ne;
}E[M];
int i,h[N],rx,ry,tot,n,m,x,y,w,head[N],cnt,f[N],t;
long long ans;
bool vis[N];
inline char gc(){
	static char buf[100000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd(){
	int x=0,fl=1;char ch=gc();
	for (;ch<48||ch>57;ch=gc())if(ch=='-')fl=-1;
	for (;48<=ch&&ch<=57;ch=gc())x=(x<<3)+(x<<1)+(ch^48);
	return x*fl;
}
bool cmp(node a,node b){
	return h[a.x]>h[b.x] || h[a.x]==h[b.x] && a.w<b.w;
}
void dfs(int u){
	vis[u]=1;t++;
	for (int i=head[u],v;i;i=E[i].ne)
		if (!vis[v=E[i].to]) dfs(v);
}
void add(int x,int y){
	E[++tot]=(kk){y,head[x]};
	head[x]=tot;
}
int find(int x){
	return x==f[x]?x:f[x]=find(f[x]);
}
int main(){
	n=rd();m=rd();
	for (i=1;i<=n;i++) h[i]=rd();
	for (i=1;i<=m;i++){
		x=e[i].x=rd();y=e[i].y=rd();w=e[i].w=rd();
		if (h[x]>=h[y]) add(x,y);
		if (h[x]<=h[y]) add(y,x);
	}
	dfs(1);
	for (i=1;i<=m;i++)
		if (vis[e[i].x] && vis[e[i].y]){
			x=e[i].x;y=e[i].y;
			if (h[x]<=h[y]) c[++cnt]=e[i];
			if (h[x]>=h[y]) c[++cnt]=(node){y,x,e[i].w};
		}
	for (i=1;i<=n;i++) f[i]=i;
	sort(c+1,c+cnt+1,cmp);
	for (i=1;i<=cnt;i++){
		rx=find(c[i].x);ry=find(c[i].y);
		if (rx!=ry){
			ans+=c[i].w;
			f[rx]=ry;
		}
	}
	printf("%d %lld",t,ans);
}