QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#556455 | #8940. Piggy Sort | ucup-team3519# | WA | 2ms | 3944kb | C++20 | 3.5kb | 2024-09-10 18:31:35 | 2024-09-10 18:31: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);
V<map<int, int>> vmp(m + 1);
V<multiset<int>> vst(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], vst[i].insert(v[i][j]), vmp[i][v[i][j]]++;
}
V<LL> sum(m + 1);
for(int i = 1; i <= m; i++) {
for(auto y : v[i]) sum[i] += y;
}
auto pos = [&](int x, int y, int pos) -> int {
if(sum[2] == sum[1]) {
if(x != y) return -1;
return x;
}
LL tmp = 1LL * (y - x) * (sum[pos] - sum[1]);
if(tmp % (sum[2] - sum[1])) return -1;
return tmp / (sum[2] - sum[1]) + x;
};
V<set<int>> posibility(n + 1);
V<map<int, V<pi>>> hav(m + 1);
for(int i = 1; i <= n; i++) {
int x = v[1][i];
for(int j = 1; j <= n; j++) {
int y = v[2][j];
bool ok = 1;
for(int k = 3; k <= m; k++) {
if(!vst[k].count(pos(x, y, k))) {
ok = 0;
break;
}
}
if(ok) {
if(posibility[i].count(y)) continue;
posibility[i].insert(y);
for(int k = 2; k <= m; k++) {
// cout << "check" << i << " " << x << " " << y << " " << k << " " << pos(x, y, k) << endl;
hav[k][pos(x, y, k)].pb({i, y});
}
}
}
}
queue<int> que;
for(int i = 1; i <= n; i++) {
if(posibility[i].size() == 1) que.push(i);
}
V<int> ansto(n + 1);
while(que.size()) {
int i = que.front();
que.pop();
int x = v[1][i];
int y = *posibility[i].begin();
// cout << "(" << x << " " << y << ")" << endl;
for(int k = 2; k <= m; k++) {
vmp[k][pos(x, y, k)]--;
if(vmp[k][pos(x, y, k)] == 0) {
// cout << "at" << k << " " << pos(x, y, k) << endl;
for(auto [a, b] : hav[k][pos(x, y, k)]) {
if(posibility[a].count(b)) {
posibility[a].erase(b);
// cout << "e " << a << " " << b << endl;
if(posibility[a].size() == 1) que.push(a);
}
}
}
}
ansto[i] = y;
}
V<int> ans(n + 1);
iota(all1(ans), 1LL);
sort(all1(ans), [&](int x, int y) -> bool {
int nowx = v[1][x];
int tox = ansto[x];
int nowy = v[1][y];
int toy = ansto[y];
if(tox - nowx != toy - nowy) return tox - nowx < toy - nowy;
return nowx < nowy;
});
// for(int i = 1; i <= n; i++) {
// cout << "i : " << i << endl;
// cout << posibility[i].size() << endl;
// for(auto x : posibility[i]) cout << x << " ";
// cout << endl;
// }
V<int> tt = ans;
for(int i = 1; i <= n; i++) {
ans[tt[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();
}
詳細信息
Test #1:
score: 100
Accepted
time: 0ms
memory: 3580kb
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: 1ms
memory: 3720kb
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: 3944kb
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: 2ms
memory: 3576kb
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: 3908kb
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: 0ms
memory: 3668kb
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:
5 4 3 2 1 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 2 3 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 1st lines differ - expected: '3 1 4 5 2', found: '5 4 3 2 1 '