USACO2014JanuaryGold

更优解法 不需要整体二分 直接并查集即可 再开一个并查集的size存1的个数
把所有的边按权值从小到大排序 一层for过去 用并查集把点并在一起
当size大于等于k时 直接res+=权值*1的个数 再把size2清0 这样就不会重复计算
比整体二分复杂度低了好多哦 然而我没想到╭(╯^╰)╮

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#define M 250005
using namespace std;
int fa[M],size[M],h[505][505],f[505][505],n,m,k;
int size2[M];
long long res;
struct nn{
    int u,to,v;
}S[M*2];
bool cmp(nn a,nn b){
    return a.v<b.v;
}
int getfa(int x){
    if(fa[x]!=x)return fa[x]=getfa(fa[x]);
    return x;
}
int main(){
    scanf("%d %d %d",&n,&m,&k);
    int tot=0;
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)scanf("%d",&h[i][j]);
    for(int i=1;i<n;i++){
        for(int j=1;j<m;j++){
        tot++;
        S[tot].u=(i-1)*m+j;S[tot].to=(i-1)*m+j+1;S[tot].v=abs(h[i][j]-h[i][j+1]);
        tot++;
        S[tot].u=(i-1)*m+j;S[tot].to=i*m+j;S[tot].v=abs(h[i][j]-h[i+1][j]);
        }
    }
    for(int j=1;j<m;j++){
        tot++;
        S[tot].u=(n-1)*m+j;S[tot].to=(n-1)*m+j+1;S[tot].v=abs(h[n][j]-h[n][j+1]);
    }
    for(int i=1;i<n;i++){
        tot++;
        S[tot].u=(i-1)*m+m;S[tot].to=(i)*m+m;S[tot].v=abs(h[i][m]-h[i+1][m]);
    }
    sort(S+1,S+tot+1,cmp);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++){
            scanf("%d",&f[i][j]);
            size2[(i-1)*m+j]=f[i][j];
        }
    for(int i=1;i<=n*m;i++)fa[i]=i,size[i]=1;
    for(int i=1;i<=tot;i++){
        int x=getfa(S[i].u);
        int y=getfa(S[i].to);
        if(x!=y){
            fa[x]=y;
            size[y]+=size[x];
            size2[y]+=size2[x];
            if(size[y]>=k){res+=1ll*size2[y]*S[i].v,size2[y]=0;}
        }
    }
    cout<<res;
    return 0;
}
/*
3 5 10
20 21 18 99 5
19 22 20 16 17
18 17 40 60 80
1 0 0 0 0
0 0 0 0 0
0 0 0 0 1
*/