【kruscal】【最小生成树】【离线】洛谷 P2266 爱的距离

建图：每个点向它四周的点连边权为两点点权的差的绝对值的边。

由于有多个需要“施法”的点，所以相当于对每个这样的点，询问与它的距离在T以内的最长边的最小值，即多次询问。

最长边最小之类的，肯定是最小生成树没跑了。BUT 若是对每个点这样做的话，肯定会TLE。

所以考虑一次处理出所有询问的答案。

在并查集将两个点集连边的时候，若两个点集的点数和<T，则对这两个集内的询问点都没有影响。

若两个点集的点数和>=T，则若A(B)集原来的点数<T，则A(B)集内的询问点都符合了题意，这个最大值就是当前这条边的边权，当然kruscal是贪心的，所以这个值是最小的。

所以并查集除了维护连通性之外，还要维护某个集合的顶点数以及某个集合的询问点数。

P.S.由于NOIP用**的devc++4.9.9.2，所以手贱地试了试，太**了，没有括号匹配，并且调试和缩进和热键都很猥琐。

P.S.P.S.运行时界面没有停留，要最后加上for(;;);，提交前千万别忘了删掉！！！！！！否则TLE得死不瞑目(<---怒立flag)。

 #include<cstdio>
 #include<algorithm>
 using namespace std;
 #define N 601
 typedef long long ll;
 int Abs(const int &x){return x< ? (-x) : x;}
 struct Edge
 {
     int u,v,w;
     Edge(const int &a,const int &b,const int &c){u=a;v=b;w=c;}
     Edge(){}
 }edges[(N*N)<<];
 bool operator < (const Edge &a,const Edge &b){return a.w<b.w;}
 int n,m,K,fa[N*N],rank[N*N],a[N][N],nm,num[N][N],en,cnt[N*N],tot,ask_sum[N*N];
 bool b[N*N];
 ll ans;
 void init(){for(int i=;i<=nm;i++) fa[i]=i,cnt[i]=,ask_sum[i]=b[i];}
 int findroot(int x)
 {
     if(x==fa[x]) return x;
     int rt=findroot(fa[x]);
     fa[x]=rt;
     return rt;
 }
 void Union(const int &U,const int &V)
 {
     if(rank[U]<rank[V]) fa[U]=V,cnt[V]+=cnt[U],ask_sum[V]+=ask_sum[U];
     else
       {
         fa[V]=U; cnt[U]+=cnt[V]; ask_sum[U]+=ask_sum[V];
         if(rank[U]==rank[V]) rank[U]++;
       }
 }
 int main()
 {
     scanf("%d%d%d",&n,&m,&K); nm=n*m;
     for(int i=;i<=n;i++)
       for(int j=;j<=m;j++)
         {
           scanf("%d",&a[i][j]);
           num[i][j]=++en;
         } en=;
     for(int i=;i<=n;i++)
       for(int j=;j<=m;j++)
         scanf("%d",&b[num[i][j]]);
     for(int i=;i<=n;i++)
       for(int j=;j<=m;j++)
         {
           if(i!=n) edges[++en]=Edge(num[i][j],num[i+][j],Abs(a[i][j]-a[i+][j]));
           if(j!=m) edges[++en]=Edge(num[i][j],num[i][j+],Abs(a[i][j]-a[i][j+]));
         }
     sort(edges+,edges+en+); init();
     for(int i=;i<=en;i++)
       {
         int f1=findroot(edges[i].u),f2=findroot(edges[i].v);
         if(f1!=f2)
           {
             if(cnt[f1]+cnt[f2]>=K)
               {
                 if(cnt[f1]<K) ans+=((ll)edges[i].w*(ll)ask_sum[f1]);
                 if(cnt[f2]<K) ans+=((ll)edges[i].w*(ll)ask_sum[f2]);
               }
             Union(f1,f2); tot++;
             if(tot==nm-) break;
           }
       }
     printf("%I64d\n",ans);
     return ;
 }