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

传送门

题意：
n个有高度的点和m条边，边只能从高点到低点走，求最小树形图？？

题解：
最小生成树。
朱刘算法求最小树形图只能得70分，考虑更高效的算法。

首先对图分层，发现低层节点对高层答案没有影响，考虑先处理高层的边。

现在假设已经处理了高层的所有边，对于本层的边，其实就是一颗最小生成树。因为高层连向本层的边看做双向边没有任何影响。那么直接把边按照层数排序，第二关键字用权值排序即可。

#include
using namespace std;
inline int read(){
    char ch=getchar();int i=0,f=1;
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){i=(i<<1)+(i<<3)+ch-'0';ch=getchar();}
    return i*f;
}
const int Maxn=1e5+50,Maxm=1e6+50;
int n,m,h[Maxn],vis[Maxn],anc[Maxn],cnt;
struct E{
    int x,y,w;
    friend inline bool operator <(const E &a,const E &b){
        return (h[a.y]>h[b.y])||(h[a.y]==h[b.y]&&a.wvector<int>e[Maxn];
inline int getanc(int x){return anc[x]==x?(x):(anc[x]=getanc(anc[x]));} 
inline void dfs(int now){
    vis[now]=1;++cnt;
    for(int o=e[now].size()-1;o>=0;o--){
        int v=e[now][o];
        if(vis[v])continue;
        dfs(v);
    }
}
int main(){
    n=read(),m=read();
    for(int i=1;i<=n;i++)h[i]=read(),anc[i]=i;
    for(int i=1;i<=m;i++){
        edge[i].x=read();edge[i].y=read();edge[i].w=read();
        if(h[edge[i].x]if(h[edge[i].y]==h[edge[i].x])e[edge[i].y].push_back(edge[i].x); 
    }
    sort(edge+1,edge+m+1);dfs(1);
    long long sum=0;
    for(int i=1;i<=m;i++){
        if(!vis[edge[i].x]||!vis[edge[i].y])continue;
        if(getanc(edge[i].x)==getanc(edge[i].y))continue;
        anc[getanc(edge[i].x)]=getanc(edge[i].y);
        sum+=edge[i].w;
    }
    cout<" "<