【洛谷P4955 】[USACO14JAN]越野滑雪越野滑雪

题目链接：ヾ(≧∇≦*)ゝ

对于每一个点可以向它右边的点和下面的点连边，权值就为两个点的高度差

然后再把所有的边按边权从小到大排序，并查集加点

最后判断当前集合是否涵盖所有的航点，如果是，就输出最后一条边的边权，否则就继续加点

Code:

#include<bits/stdc++.h>
#define N 500001
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
int n,m,t,cnt,tt;ll ans;
int fa[N],ct[501][501];
int mp[501][501],sz[N],u[N];
struct dqy{ll v;int x,y;}w[N];
inline bool cmp(dqy a,dqy b){return a.v<b.v;}
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
inline pii getpos(int num){
    int x=num/m+(num%m!=0);
    int y=num-(x-1)*m;
    return make_pair(x,y);
}
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-48;ch=getchar();}
    return x*f;
}
signed main(){
    n=read(),m=read();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            mp[i][j]=read();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            ct[i][j]=read(),tt+=ct[i][j];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++){
            if(j!=m){
                w[++cnt].x=(i-1)*m+j;
                w[cnt].y=w[cnt].x+1;
                w[cnt].v=abs(mp[i][j]-mp[i][j+1]);
            }
            if(i!=n){
                w[++cnt].x=(i-1)*m+j;
                w[cnt].y=w[cnt].x+m;
                w[cnt].v=abs(mp[i][j]-mp[i+1][j]);
            }
        }
    sort(w+1,w+cnt+1,cmp);
    for(int i=1;i<=n*m;i++) fa[i]=i;
    for(int i=1;i<=n*m;i++){
        sz[i]=1;pii now=getpos(i);
        u[i]=ct[now.first][now.second];
    }
    for(int i=1;i<=cnt;i++){
        int x=w[i].x,y=w[i].y;
        int f1=find(x),f2=find(y);
        if(f1==f2) continue;
        sz[f1]+=sz[f2];
        u[f1]+=u[f2];fa[f2]=f1;
        if(u[f1]==tt){
            ans=w[i].v;
            break;
        }
    }
    printf("%lld\n",ans);
    return 0;
}

转载于:https://www.cnblogs.com/NLDQY/p/10467537.html