結果

問題 No.2616 中央番目の中央値
コンテスト
ユーザー Carpenters-Cat
提出日時 2024-01-27 02:25:54
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 181 ms / 2,000 ms
コード長 941 bytes
コンパイル時間 4,234 ms
コンパイル使用メモリ 253,836 KB
最終ジャッジ日時 2025-02-19 00:05:01
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 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 a = pre[i];
		int d = aft[i];
		int b = iv[i] - a;
		int c = (N - iv[i] - 1) - d;
		ans += kai[a + d] * fkai[a] * fkai[d] * kai[b + c] * fkai[b] * fkai[c];
	}
	cout << ans.val() << endl;
}
0