QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#841945 | #9967. Imbalanced Teams | ucup-team4893# | WA | 14ms | 27164kb | C++17 | 1.8kb | 2025-01-04 08:58:37 | 2025-01-04 08:58:45 |
Judging History
answer
#include<iostream>
#include<cstdio>
#include<fstream>
#include<queue>
#include<algorithm>
#define fopen(x, y) freopen(x".in", "r", stdin); freopen(y".out", "w", stdout);
#define int long long
#ifdef int
#define inf 0x3f3f3f3f3f3f3f3fll
#else
#define inf 0x3f3f3f3f
#endif
using namespace std;
const int mod = 998244353;
int n, k, a[2005], pn, fac[1000005], ifac[1000005], inv[1000005];
pair<int, int> p[2005];
inline int C(int x, int y) {
if(y > x || y < 0) return 0;
return fac[x] * ifac[y] % mod * ifac[x - y] % mod;
}
void work() {
cin >> n >> k;
for(int i = 1; i <= n; i++) {
cin >> a[i];
}
sort(a + 1, a + n + 1);
int sum = 0;
for(int i = 1; i <= n; i++) {
if(a[i] != a[i - 1]) pn++;
p[pn].first = a[i]; p[pn].second++;
if(i <= k) sum -= a[i];
if(i >= n - k + 1) sum += a[i];
}
cout << sum << ' ';
int kl = k, kr = k, pl = 0, pr = 0;
for(int i = 1; i <= n; i++) {
if(kl >= p[i].second) kl -= p[i].second;
else {
pl = i;
break;
}
}
for(int i = n; i >= 1; i--) {
if(kr >= p[i].second) kr -= p[i].second;
else {
pr = i;
break;
}
}
if(pl == pr) {
if(kl == k && kr == k) {
cout << C(p[pl].second, kl) * C(p[pr].second - kl, kr) % mod * (mod + 1) / 2 % mod;
}
else cout << C(p[pl].second, kl) * C(p[pr].second - kl, kr) % mod << '\n';
}
else {
// cout << p[pl].second << ' ' << kl << ' ' << p[pr].second << ' ' << kr << endl;
cout << C(p[pl].second, kl) * C(p[pr].second, kr) % mod << '\n';
}
}
signed main() {
ios::sync_with_stdio(false); cin.tie(0);
int _ = 1;
fac[0] = fac[1] = ifac[0] = ifac[1] = inv[1] = 1;
for(int i = 2; i <= 1000000; i++) {
inv[i] = (mod - mod / i) * inv[mod % i] % mod;
fac[i] = fac[i - 1] * i % mod;
ifac[i] = ifac[i - 1] * inv[i] % mod;
}
// cin >> _;
while(_--) work();
}
详细
Test #1:
score: 100
Accepted
time: 4ms
memory: 27088kb
input:
6 2 2 5 7 2 5 2
output:
8 6
result:
ok 2 number(s): "8 6"
Test #2:
score: 0
Accepted
time: 6ms
memory: 27032kb
input:
5 2 1 1 1 1 1
output:
0 15
result:
ok 2 number(s): "0 15"
Test #3:
score: 0
Accepted
time: 8ms
memory: 27080kb
input:
2 1 1 1
output:
0 1
result:
ok 2 number(s): "0 1"
Test #4:
score: 0
Accepted
time: 7ms
memory: 27164kb
input:
2 1 1 2
output:
1 1
result:
ok 2 number(s): "1 1"
Test #5:
score: 0
Accepted
time: 14ms
memory: 27080kb
input:
10 4 3 3 1 2 4 6 2 4 4 1
output:
12 1
result:
ok 2 number(s): "12 1"
Test #6:
score: 0
Accepted
time: 13ms
memory: 27092kb
input:
14 3 57 57 57 57 57 57 57 57 57 57 57 57 57 57
output:
0 30030
result:
ok 2 number(s): "0 30030"
Test #7:
score: 0
Accepted
time: 7ms
memory: 27160kb
input:
13 5 858336 900782 858336 900782 900782 858336 900782 858336 858336 858336 858336 858336 52093
output:
976027 280
result:
ok 2 number(s): "976027 280"
Test #8:
score: 0
Accepted
time: 6ms
memory: 27096kb
input:
14 4 447923 447923 447923 211106 447923 447923 447923 447923 16966 447923 211106 515550 211106 211106
output:
1209035 224
result:
ok 2 number(s): "1209035 224"
Test #9:
score: 0
Accepted
time: 7ms
memory: 27084kb
input:
2000 935 57596 988638 30477 956599 52986 460052 987863 291984 947848 109335 541003 338365 939256 297365 926486 944912 700042 810595 412192 37130 343207 311311 681629 48155 319677 435667 731251 919378 254216 893282 661237 740159 787502 501360 517533 349880 565298 536545 192793 18666 425164 856977 536...
output:
493241703 1
result:
ok 2 number(s): "493241703 1"
Test #10:
score: -100
Wrong Answer
time: 3ms
memory: 27108kb
input:
2000 1000 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101493 101...
output:
0 236399791
result:
wrong answer 2nd numbers differ - expected: '36237869', found: '236399791'