// PA2014, runda 5, Ciagi
// Andrzej Pezarski
#include <cstdio>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
using namespace std;
int N, K;
pair<int, int> T[100000][5];
int X[100000];
pair<long long, int> D[5];
int main() {
scanf("%d%d", &N, &K);
K=3;
for (int k=0; k<K; k++) {
D[k]=make_pair(0LL, k);
for (int i=0; i<N; i++) {
int c;
scanf("%d", &c);
T[i][k]=make_pair(c, k);
}
}
for (int i=0; i<N; i++) {
sort(T[i], T[i]+K);
X[i]=T[i][1].first;
}
for (int i=0; i<N; i++) {
for (int k=0; k<K; k++) {
D[T[i][k].second].first+=abs(T[i][k].first-X[i]);
}
}
sort(D, D+K);
long long d=(D[K-1].first-D[K-2].first)/2;
int a=D[K-1].second;
for (int i=0; i<N; i++) {
if (T[i][0].second==a) {
long long dd=T[i][1].first-T[i][0].first;
dd=min(dd, d);
X[i]-=dd;
d-=dd;
}
if (T[i][2].second==a && K==3) {
long long dd=T[i][2].first-T[i][1].first;
dd=min(dd, d);
X[i]+=dd;
d-=dd;
}
}
for (int i=0; i<N; i++) {
printf("%d ", X[i]);
}
printf("\n");
// for (int k=0; k<K; k++)
// D[k]=make_pair(0LL, k);
//
// for (int i=0; i<N; i++) {
// for (int k=0; k<K; k++) {
// D[T[i][k].second].first+=abs(T[i][k].first-X[i]);
// }
// }
// sort(D, D+K);
// printf("%lld\n", D[K-1].first);
//
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 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | // PA2014, runda 5, Ciagi // Andrzej Pezarski #include <cstdio> #include <cstdlib> #include <vector> #include <set> #include <map> #include <algorithm> using namespace std; int N, K; pair<int, int> T[100000][5]; int X[100000]; pair<long long, int> D[5]; int main() { scanf("%d%d", &N, &K); K=3; for (int k=0; k<K; k++) { D[k]=make_pair(0LL, k); for (int i=0; i<N; i++) { int c; scanf("%d", &c); T[i][k]=make_pair(c, k); } } for (int i=0; i<N; i++) { sort(T[i], T[i]+K); X[i]=T[i][1].first; } for (int i=0; i<N; i++) { for (int k=0; k<K; k++) { D[T[i][k].second].first+=abs(T[i][k].first-X[i]); } } sort(D, D+K); long long d=(D[K-1].first-D[K-2].first)/2; int a=D[K-1].second; for (int i=0; i<N; i++) { if (T[i][0].second==a) { long long dd=T[i][1].first-T[i][0].first; dd=min(dd, d); X[i]-=dd; d-=dd; } if (T[i][2].second==a && K==3) { long long dd=T[i][2].first-T[i][1].first; dd=min(dd, d); X[i]+=dd; d-=dd; } } for (int i=0; i<N; i++) { printf("%d ", X[i]); } printf("\n"); // for (int k=0; k<K; k++) // D[k]=make_pair(0LL, k); // // for (int i=0; i<N; i++) { // for (int k=0; k<K; k++) { // D[T[i][k].second].first+=abs(T[i][k].first-X[i]); // } // } // sort(D, D+K); // printf("%lld\n", D[K-1].first); // return 0; } |
English