CodeForces 650C Table Compression

题目链接: http://codeforces.com/problemset/problem/650/C

---------------------------------------------------------------------------------------------------------

首先要知道的是这题原本相同的数最后是可以不同的

然后主要是要考虑到$Z$字形的数据 ($a$代表某相同数字 $x$代表其他数字)

$aax$

$xaa$

这种情况下第一行第一个和第二行第三个是必须相同的

如果只是对行和列进行简单的十字型的判断的话这两个位置最终的数字很有可能不同

所以我们要对行和列的当前最大数的最小值进行维护 然后就是写一个并查集了

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 1e6 + 10;
vector <int> a[N], b[N];
int s[N], num[N << 1], fa[N];
int n, m, len, ans;
int findf(int x)
{
    if(fa[x] != x)
        fa[x] = findf(fa[x]);
    return fa[x];
}
void merge(int x, int y)
{
    int tx = findf(x), ty = findf(y);
    if(tx != ty)
    {
        if(num[tx] >= num[ty])
            fa[ty] = tx;
        else
            fa[tx] = ty;
    }
}
int main()
{
    int x, y;
    scanf("%d%d", &n, &m);
    for(int i = 0; i < n; ++i)
        for(int j = 0; j < m; ++j)
        {
            scanf("%d", &x);
            a[i].push_back(x);
            s[len++] = x;
        }
    sort(s, s + len);
    len = unique(s, s + len) - s;
    for(int i = 0; i < n; ++i)
        for(int j = 0; j < m; ++j)
        {
            a[i][j] = lower_bound(s, s + len, a[i][j]) - s;
            b[a[i][j]].push_back(i * m + j);
        }
    for(int i = 0; i < n + m; ++i)
        fa[i] = i;
    for(int i = 0; i < len; ++i)
    {
        for(int j = 0, sz = b[i].size(); j < sz; ++j)
        {
            x = b[i][j] / m;
            y = b[i][j] % m;
            merge(x, n + y);
        }
        for(int j = 0, sz = b[i].size(); j < sz; ++j)
        {
            x = b[i][j] / m;
            y = b[i][j] % m;
            a[x][y] = num[findf(x)] + 1;
        }
        for(int j = 0, sz = b[i].size(); j < sz; ++j)
        {
            x = b[i][j] / m;
            y = b[i][j] % m;
            num[x] = max(num[x], a[x][y]);
            num[n + y] = max(num[n + y], a[x][y]);
            fa[x] = x;
            fa[n + y] = n + y;
        }
    }
    for(int i = 0; i < n; ++i)
        for(int j = 0; j < m; ++j)
            printf("%d%c", a[i][j], (j == m - 1 ? '
' : ' '));
    return 0;
}