QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#558912 | #8940. Piggy Sort | ucup-team3519 | WA | 1ms | 3836kb | C++20 | 2.6kb | 2024-09-11 19:16:35 | 2024-09-11 19:16:36 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define pb push_back
#define all1(x) (x).begin() + 1, (x).end()
#define all0(x) (x).begin(), (x).end()
typedef long long LL;
typedef pair<int, int> pi;
void solve() {
int n; cin >> n;
int m; cin >> m;
V<V<int>> v(m + 1);
for(int i = 1; i <= m; i++) {
v[i].resize(n + 1);
for(int j = 1; j <= n; j++) cin >> v[i][j];
}
V<LL> sumini(m + 1);
V<int> ord(m + 1);
iota(all1(ord), 1);
for(int i = 1; i <= m; i++) sumini[i] = accumulate(all1(v[i]), 0LL);
sort(all1(ord), [&](int x, int y) {
return sumini[x] < sumini[y];
});
map<int, int> id;
for(int i = 1; i <= n; i++) {
id[v[ord[1]][i]] = i;
}
V<multiset<int>> vst(m + 1);
for(int i = 1; i <= m; i++) {
for(int j = 1; j <= n; j++) vst[i].insert(v[ord[i]][j]);
}
V<LL> sum(m + 1);
for(int i = 1; i <= m; i++) {
sum[i] = accumulate(all1(v[ord[i]]), 0LL);
}
auto get_mx = [&](multiset<int> & st) {
return *st.rbegin();
};
auto pos = [&](int p1, int x, int y, int pos) -> LL {
if(sum[2] == sum[1]) {
if(x != y) return -1;
return x;
}
LL tmp = 1LL * (y - x) * (sum[pos] - sum[p1]);
if(tmp % (sum[p1 + 1] - sum[p1])) return -1;
return tmp / (sum[p1 + 1] - sum[p1]) + x;
};
V<int> ans(n + 1);
for(int i = 1; i <= n; i++) {
int x = -1, y = 2;
for(int j = 2; j < m; j++) {
int p = pos(j - 1, get_mx(vst[j - 1]), get_mx(vst[j]), j + 1);
if(p == get_mx(vst[j + 1])) {
x = j, y = j + 1;
break;
}
}
if(x == -1) {
for(int i = 1; i <= n; i++) cout << n - i + 1 << " ";
cout << endl;
return;
}
int vx = get_mx(vst[x]), vy = get_mx(vst[y]);
int a = pos(x, get_mx(vst[x]), get_mx(vst[y]), 1);
int b = pos(x, get_mx(vst[x]), get_mx(vst[y]), 2);
ans[id[a]] = b - a;
for(int i = 1; i <= m; i++) {
vst[i].erase(vst[i].find(pos(x, vx, vy, i)));
}
}
V<int> ansord(n + 1);
iota(all1(ansord), 1);
stable_sort(all1(ansord), [&](int x, int y) -> bool {
return ans[x] < ans[y];
});
for(int i = 1; i <= n; i++) ans[ansord[i]] = i;
for(int i = 1; i <= n; i++) cout << ans[i] << " ";
cout << endl;
}
int main() {
ios::sync_with_stdio(0), cin.tie(0);
int t; cin >> t;
while(t--) solve();
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3544kb
input:
3 2 4 1 2 3 4 5 6 7 8 1 2 1 1 3 4 1 2 3 6 9 9 10 15 17 12 18 21
output:
1 2 1 3 1 2
result:
ok 3 lines
Test #2:
score: 0
Accepted
time: 0ms
memory: 3580kb
input:
41 1 2 -19 9531 2 3 11 13 3175 4759 2211 3313 10 19 -54 -25 -19 -18 -1 3 61 63 85 88 -54 753 863 2397 3111 4649 4671 4756 5507 7762 -54 369 479 1245 1575 2345 2367 2452 2819 3922 -54 553 663 1797 2311 3449 3471 3556 4107 5762 -54 87 197 399 447 653 675 760 845 1102 -54 320 430 1098 1379 2051 2073 21...
output:
1 1 2 1 2 6 10 5 7 9 4 3 8 8 7 5 9 2 1 6 3 4 1 6 5 9 8 2 3 10 4 7 3 5 10 6 7 4 9 8 2 1 4 3 1 8 2 5 6 7 3 1 5 2 6 9 8 7 4 10 2 3 1 4 3 2 9 1 4 6 7 5 8 1 5 2 6 7 3 4 8 9 9 8 5 7 2 3 4 6 1 1 2 4 8 5 7 10 6 9 3 7 1 3 2 9 8 10 4 5 6 7 8 1 2 5 6 3 9 4 8 1 2 7 6 9 3 5 4 8 1 2 4 5 10 9 3 6 7...
result:
ok 41 lines
Test #3:
score: 0
Accepted
time: 1ms
memory: 3572kb
input:
46 8 12 -50 -35 -20 -10 -9 4 13 91 -24 30 32 143 146 147 173 221 -44 -8 10 13 26 27 61 103 -46 -12 -3 8 14 15 45 99 -22 32 36 147 158 159 189 237 -47 -14 -11 7 8 9 37 97 -31 18 23 104 105 117 129 165 -45 -10 5 9 20 21 53 101 -36 8 18 74 75 77 119 125 -30 20 24 110 111 125 131 173 -48 -19 -16 2 3 6 2...
output:
1 7 3 5 6 2 8 4 1 2 3 3 6 1 5 10 8 7 2 9 4 2 5 3 6 9 10 8 7 1 4 1 4 5 2 6 3 7 8 6 4 5 7 9 2 3 1 6 7 2 8 9 1 10 3 5 4 4 5 1 3 2 1 2 1 2 4 5 7 6 3 8 9 6 3 4 9 7 1 2 8 10 5 1 6 7 8 4 9 5 3 2 1 4 2 6 5 8 3 7 2 1 5 1 7 6 3 2 4 7 3 8 4 1 2 9 5 6 10 6 1 2 5 3 4 2 1 3 4 7 8 5 9 2 1 4 3 6 ...
result:
ok 46 lines
Test #4:
score: 0
Accepted
time: 1ms
memory: 3672kb
input:
48 3 4 -4952 -1539 836 -4294 5909 12778 -4811 57 3395 -4529 3249 8513 8 11 -9107 -1143 1324 3936 4088 4381 7658 9440 -2753 531 6032 14986 18097 18264 20240 22022 -5224 -120 5276 9673 12692 12763 15347 17129 -2047 717 6248 16504 19621 19856 21638 23420 -6283 -399 4952 7396 10304 10477 13250 15032 -48...
output:
1 2 3 3 1 6 8 2 7 4 5 8 4 6 1 2 9 10 5 3 7 1 2 3 6 1 10 7 5 3 9 4 8 2 1 2 8 3 2 7 10 6 1 9 5 4 9 5 2 7 6 1 8 4 3 6 4 3 2 7 5 1 1 4 2 3 1 3 8 9 6 7 4 1 2 10 5 5 1 6 4 2 3 2 1 9 6 4 5 3 8 7 3 5 6 7 9 2 4 8 1 4 2 1 3 5 6 4 5 6 3 7 1 2 1 9 7 10 1 8 5 3 6 2 4 8 3 6 9 2 5 4 7 1 4 3 9 5...
result:
ok 48 lines
Test #5:
score: 0
Accepted
time: 1ms
memory: 3540kb
input:
40 10 20 -4289879 -3663596 -3442064 -3379220 -670906 -329052 1547135 1640345 2662172 3577480 -4280827 -3609576 -3374758 -3321039 -598417 -319197 1583489 1685532 2700424 3645662 -4276115 -3581456 -3339722 -3290753 -560683 -314067 1602413 1709054 2720336 3681154 -4271279 -3552596 -3303764 -3259670 -52...
output:
1 6 8 7 10 2 3 5 4 9 2 3 1 9 8 3 6 4 2 10 1 7 5 2 1 8 5 3 4 10 7 6 9 7 2 6 1 9 5 8 3 4 2 6 3 9 7 1 4 5 8 3 6 7 5 1 2 8 4 3 4 10 9 8 7 6 2 5 1 3 8 2 6 1 4 5 7 4 2 5 7 1 6 3 1 8 3 10 9 7 4 6 5 2 1 3 5 4 2 8 6 7 3 1 4 5 10 2 9 1 2 3 5 10 2 7 6 4 1 8 3 9 4 10 7 6 8 9 5 1 2 3 5 7 10 1 4 9...
result:
ok 40 lines
Test #6:
score: -100
Wrong Answer
time: 1ms
memory: 3836kb
input:
79 5 7 -5 -1 7 8 10 -1 1 10 19 20 -1 4 10 25 26 -1 5 10 27 28 -2 -1 10 13 14 -4 -1 9 10 10 -1 3 10 23 24 5 6 -7 -5 2 3 5 2 3 11 13 14 -5 -3 2 3 6 2 3 7 9 12 2 3 3 5 10 2 3 5 7 11 3 4 -10 -7 -5 -10 1 3 -10 -4 -2 -10 0 2 5 10 -10 -6 -1 6 7 -10 4 7 9 11 -10 7 8 13 13 -10 7 14 16 19 -10 7 12 15 17 -10 6...
output:
3 1 4 5 2 4 5 1 2 3 1 2 3 1 4 5 3 2 3 2 1 1 3 4 5 1 2 3 4 1 5 2 1 4 2 5 3 2 3 4 5 1 3 2 1 1 1 3 4 5 2 1 3 2 4 5 5 2 1 3 4 1 3 4 5 2 2 1 3 4 5 1 4 2 3 5 4 1 2 3 1 4 2 3 1 1 3 4 5 2 1 3 4 2 1 2 3 1 1 2 3 4 5 2 1 3 4 5 1 2 1 2 3 5 4 1 2 3 2 5 1 3 4 1 2 4 3 5 1 2 1 3 4 5...
result:
wrong answer 11th lines differ - expected: '2 3 1', found: '3 2 1 '