結果

問題 No.2616 中央番目の中央値
ユーザー Carpenters-CatCarpenters-Cat
提出日時 2024-01-27 01:44:36
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,003 bytes
コンパイル時間 4,869 ms
コンパイル使用メモリ 265,244 KB
実行使用メモリ 12,672 KB
最終ジャッジ日時 2024-09-28 09:27:02
合計ジャッジ時間 11,507 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 37
権限があれば一括ダウンロードができます

ソースコード

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