Codeforces 650C Table Compression (并查集)

题意：M×N的矩阵 让你保持每行每列的大小对应关系不变，将矩阵重写，重写后的最大值最小。

思路：离散化思想+并查集，详见代码

好题！

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <bits/stdc++.h>
using namespace std;
#define clc(a,b) memset(a,b,sizeof(a))
const double pi=acos(-1);
const int maxn=1000010;
int k,n,m,tot,i,j;
struct data {
    int r,c,v,id;
} p[maxn];
bool cmp(const data &a,const data &b) {
    return a.v<b.v;
}
int f[maxn],X[maxn],Y[maxn],x[maxn],y[maxn],ans[maxn],tmp[maxn];

int F(int a) {
    return a==f[a]?f[a]:f[a]=F(f[a]);
}

int main() {
    //freopen("in.txt","r",stdin);
    scanf("%d%d",&n,&m);
    tot=0;
    clc(tmp,0);
    clc(f,-1);
    for( i=1; i<=n; i++) {
        for( j=1; j<=m; j++) {
            scanf("%d",&p[++tot].v);
            p[tot].r=i,p[tot].c=j;
            p[tot].id=tot;
            f[tot]=tot;
        }
    }
    sort(p+1,p+1+tot,cmp);
    for(i=1; i<=tot; i=j) {
        for(j=i; p[i].v==p[j].v; j++);//每次只处理相同元素
        for(k=i; k<j; k++) {//每行每列并查集统计，归并到一个集合
            int r=p[k].r,c=p[k].c;
            if(!x[r]) x[r]=k;
            else
                f[F(k)]=F(x[r]);
            if(!y[c]) y[c]=k;
            else
                f[F(k)]=F(y[c]);
        }
        for(k=i; k<j; k++) {//当前值的改变值 应该是该元素所在行或列 前一个填入的值再加一
            int q=F(k);
            tmp[q]=max(tmp[q],max(X[p[k].r],Y[p[k].c])+1);
        }
        for(k=i; k<j; k++) {//X Y数组维护行和列的最大填入值
            x[p[k].r]=y[p[k].c]=0;
            X[p[k].r]=Y[p[k].c]=ans[p[k].id]=tmp[F(k)];
        }
    }
    for(i=1; i<=tot; i++) {
        if(i%m==1)
            printf("%d",ans[i]);
        else
            printf(" %d",ans[i]);
        if(i%m==0)
            printf("\n");
    }
    return 0;
}