結果

問題 No.2616 中央番目の中央値
ユーザー Carpenters-CatCarpenters-Cat
提出日時 2024-01-27 01:44:36
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,003 bytes
コンパイル時間 4,433 ms
コンパイル使用メモリ 266,432 KB
実行使用メモリ 12,800 KB
最終ジャッジ日時 2024-01-27 01:44:47
合計ジャッジ時間 11,396 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
int main () {
	int N;
	cin >> N;
	std::vector<int> P(N), iv(N);
	for (int i = 0; i < N; i ++) {
		cin >> P[i];
		iv[--P[i]] = i;
	}
	fenwick_tree<int> F1(N), F2(N);
	vector<mint> kai(N + 1, mint(1)), fkai(N + 1, mint(1));
	for (int i = 1; i <= N; i ++) {
		kai[i] = kai[i - 1] * i;
	}
	fkai[N] = fkai[N] / kai[N];
	for (int i = N; i; i --) {
		fkai[i - 1] = fkai[i] * i;
	}
	mint ans = 0;
	vector<int> pre(N), aft(N);
	for (int i = 0; i < N; i ++) {
		pre[i] = F1.sum(0, iv[i]);
		F1.add(iv[i], 1);
	}
	for (int i = N - 1; i >= 0; i --) {
		aft[i] = F2.sum(iv[i], N);
		F2.add(iv[i], 1);
	}
	for (int i = 0; i < N; i ++) {
		int p = pre[i];
		int a = aft[i];
		int n = p + a, r = min(p, a);
		ans += kai[n] / fkai[r] / fkai[n - r];
		p = iv[i] - p;
		a = (N - iv[i] - 1) - a;
		n = p + a;
		r = min(p, a);
		ans += kai[n] / fkai[r] / fkai[n - r];
	}
	cout << ans.val() << endl;
}
0