Codeforces #345 div1 C. Table Compression 并查集 构造

题目

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

题目来源：http://codeforces.com/contest/650

简要题意：给定一个矩阵，令矩阵元素非负，最大元素最小，每行每列元素相对大小不变。

题解

利用并查集连通每行每列相同的元素。

对矩阵元素按照值排序，从小到大不断询问该行该列的情况，更新自身，再去更新每行每列的情况。

由于升序，所以每行每列最后更新的值都会是小于等于当前元素的。

根据并查集中记录的信息可以进行判断。

需要注意的是应该记录的是编号而非值，因为一个数字对应的值会更新。

比赛的时候最后时刻想到并查集，但是已经来不及了，思路基本还是对的。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
// head
const int N = 1e6+5;

struct DisjointSetUnion {
    int fa[N];
    int v[N];

    int find(int key) {
        return (key == fa[key]) ? key : (fa[key] = find(fa[key]));
    }
    void init(int n) {
        for (int i = 0; i <= n; i++) {
            fa[i] = i;
            v[i] = 0;
        }
    }
    void joint(int u, int v) {
        u = find(u), v = find(v);
        if (u != v) {
            fa[u] = v;
        }
    }
    bool same(int u, int v) {
        return find(u) == find(v);
    }
    void update(int no, int va) {
        no = find(no);
        v[no] = max(v[no], va);
    }
    int query(int no) {
        no = find(no);
        return v[no];
    }
};


struct Node {
    int v, no;
    PII p;
};
bool cmp(const Node &a, const Node &b) {
    return a.v < b.v;
}
bool cmp2(const Node &a, const Node &b) {
    return a.p < b.p;
}

Node a[N];
DisjointSetUnion dsu;

int row[N], col[N];

void buildDsu(int n, int m) {
    int tot = n * m;
    for (int i = 0; i < m; i++) {
        map<int, int> ma;
        for (int j = i; j < tot; j += m) {
            if (ma.find(a[j].v) != ma.end()) {
                dsu.joint(a[j].no, ma[a[j].v]);
            }
            ma[a[j].v] = a[j].no;
        }
    }
    for (int i = 0; i < tot; i += m) {
        map<int, int> ma;
        for (int j = i; j < i+m; j++) {
            if (ma.find(a[j].v) != ma.end()) {
                dsu.joint(a[j].no, ma[a[j].v]);
            }
            ma[a[j].v] = a[j].no;
        }
    }
}

void update(Node &cur) {
    int r = cur.p.fi, c = cur.p.se;
    int ans = 0;
    if (!dsu.same(cur.no, row[r])) ans = max(ans, dsu.query(row[r]) + 1);
    if (!dsu.same(cur.no, col[c])) ans = max(ans, dsu.query(col[c]) + 1)
    dsu.update(cur.no, ans);
    row[r] = col[c] = cur.no;
}

int main() {
    int n, m, cnt = 0;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) {
            scanf("%d", &a[cnt].v);
            a[cnt].no = cnt+1;
            a[cnt++].p = make_pair(i, j);
        }
    }
    dsu.init(cnt);
    buildDsu(n, m);
    sort(a, a + cnt, cmp);

    for (int i = 0; i < cnt; i++) {
        update(a[i]);
    }

    sort(a, a+cnt, cmp2);
    for (int i = 0; i < cnt; i++) {
        printf("%d%c", dsu.query(a[i].no), a[i].p.se == m ? '\n' : ' ');
    }
    return 0;
}