#include <iostream>
#include <vector>
#include <cmath>
#include <cassert>
using namespace std;
const int MAXN = (1<<17) + 1;
const int MAXK = 6;
int a[MAXK][MAXN];
int d[MAXK];
int n, k;
int main() {
assert(2 == scanf("%d%d", &n, &k));
for (int j = 0; j < k; ++j) {
d[j] = 0;
for (int i = 0; i < n; ++i) {
assert(1 == scanf("%d", &a[j][i]));
}
}
for (int i = 0; i < n; ++i) {
int jmin = 0;
int jmax = 0;
for (int j = 0; j < k; ++j) {
if (a[j][i] < a[jmin][i] or (a[j][i] == a[jmin][i] and d[j] > d[jmin])) {
jmin = j;
}
if (a[j][i] > a[jmax][i] or (a[j][i] == a[jmax][i] and d[j] > d[jmax])) {
jmax = j;
}
}
int x = min(a[jmax][i], max(a[jmin][i], (a[jmax][i] + a[jmin][i] + d[jmax] - d[jmin]) / 2));
for (int j = 0; j < k; ++j) {
d[j] += abs(a[j][i] - x);
}
printf("%d ", x);
}
printf("\n");
/*for (int i = 0; i < k; ++i) {
printf("%d: %d\n", i, d[i]);
}*/
return 0;
}
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 | #include <iostream> #include <vector> #include <cmath> #include <cassert> using namespace std; const int MAXN = (1<<17) + 1; const int MAXK = 6; int a[MAXK][MAXN]; int d[MAXK]; int n, k; int main() { assert(2 == scanf("%d%d", &n, &k)); for (int j = 0; j < k; ++j) { d[j] = 0; for (int i = 0; i < n; ++i) { assert(1 == scanf("%d", &a[j][i])); } } for (int i = 0; i < n; ++i) { int jmin = 0; int jmax = 0; for (int j = 0; j < k; ++j) { if (a[j][i] < a[jmin][i] or (a[j][i] == a[jmin][i] and d[j] > d[jmin])) { jmin = j; } if (a[j][i] > a[jmax][i] or (a[j][i] == a[jmax][i] and d[j] > d[jmax])) { jmax = j; } } int x = min(a[jmax][i], max(a[jmin][i], (a[jmax][i] + a[jmin][i] + d[jmax] - d[jmin]) / 2)); for (int j = 0; j < k; ++j) { d[j] += abs(a[j][i] - x); } printf("%d ", x); } printf("\n"); /*for (int i = 0; i < k; ++i) { printf("%d: %d\n", i, d[i]); }*/ return 0; } |
English