QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#841945#9967. Imbalanced Teamsucup-team4893#WA 14ms27164kbC++171.8kb2025-01-04 08:58:372025-01-04 08:58:45

Judging History

你现在查看的是最新测评结果

  • [2025-01-04 08:58:45]
  • 评测
  • 测评结果:WA
  • 用时:14ms
  • 内存:27164kb
  • [2025-01-04 08:58:37]
  • 提交

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();
}

Details

Tip: Click on the bar to expand more detailed information

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'