-
Notifications
You must be signed in to change notification settings - Fork 184
/
Copy pathb.cc
51 lines (49 loc) · 1.15 KB
/
b.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
// https://codeforces.com/contest/1517/problem/B
#include <bits/stdc++.h>
using namespace std;
using vi = vector<int>;
using vvi = vector<vi>;
using iii = tuple<int, int, int>;
using viii = vector<iii>;
int main() {
cin.tie(0), ios::sync_with_stdio(0);
int t, n, m;
cin >> t;
while (t--) {
cin >> n >> m;
vvi b(n, vi(m)), c(n, vi(m, -1)), d(m, vi(n, -1));
viii e;
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++) {
cin >> b[i][j];
e.push_back({b[i][j], i, j});
}
sort(e.begin(), e.end());
for (int i = 0; i < m; i++) {
int u, v, w;
tie(w, u, v) = e[i];
c[u][v] = i;
d[i][u] = v;
}
int k = m;
vvi f(n);
for (int i = m; i < n * m; i++) {
int u, v, w;
tie(w, u, v) = e[k++];
f[u].push_back(v);
}
vi ks(n);
for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++)
if (d[i][j] < 0) {
int v = f[j][ks[j]++];
c[j][v] = i;
d[i][j] = v;
}
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++) {
int k = d[j][i];
cout << b[i][k] << " \n"[j == m - 1];
}
}
}