結果

問題 No.1300 Sum of Inversions
ユーザー 👑 ygussanyygussany
提出日時 2020-11-27 23:52:59
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 98 ms / 2,000 ms
コード長 1,826 bytes
コンパイル時間 401 ms
コンパイル使用メモリ 33,596 KB
実行使用メモリ 9,604 KB
最終ジャッジ日時 2024-07-26 20:13:42
合計ジャッジ時間 4,223 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,908 KB
testcase_01 AC 3 ms
7,412 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 74 ms
8,904 KB
testcase_04 AC 69 ms
9,044 KB
testcase_05 AC 57 ms
8,460 KB
testcase_06 AC 83 ms
9,260 KB
testcase_07 AC 78 ms
9,088 KB
testcase_08 AC 91 ms
9,432 KB
testcase_09 AC 88 ms
9,304 KB
testcase_10 AC 48 ms
8,136 KB
testcase_11 AC 49 ms
8,280 KB
testcase_12 AC 73 ms
9,084 KB
testcase_13 AC 71 ms
8,984 KB
testcase_14 AC 98 ms
9,568 KB
testcase_15 AC 90 ms
9,312 KB
testcase_16 AC 78 ms
8,888 KB
testcase_17 AC 46 ms
8,056 KB
testcase_18 AC 53 ms
8,300 KB
testcase_19 AC 61 ms
8,744 KB
testcase_20 AC 64 ms
8,872 KB
testcase_21 AC 62 ms
8,636 KB
testcase_22 AC 58 ms
8,992 KB
testcase_23 AC 81 ms
9,268 KB
testcase_24 AC 59 ms
8,520 KB
testcase_25 AC 51 ms
8,840 KB
testcase_26 AC 49 ms
8,560 KB
testcase_27 AC 54 ms
8,540 KB
testcase_28 AC 88 ms
9,492 KB
testcase_29 AC 63 ms
8,840 KB
testcase_30 AC 86 ms
9,348 KB
testcase_31 AC 58 ms
8,740 KB
testcase_32 AC 59 ms
9,076 KB
testcase_33 AC 30 ms
9,600 KB
testcase_34 AC 40 ms
9,604 KB
testcase_35 AC 59 ms
9,592 KB
testcase_36 AC 62 ms
9,488 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>

const int Mod = 998244353;

typedef struct {
	int key, id;
} data;

void merge_sort(data x[], int n)
{
	static data y[200001] = {};
	if (n <= 1) return;
	merge_sort(&(x[0]), n/2);
	merge_sort(&(x[n/2]), (n+1)/2);
	
	int i, p, q;
	for (i = 0, p = 0, q = n/2; i < n; i++) {
		if (p >= n/2) y[i] = x[q++];
		else if (q >= n) y[i] = x[p++];
		else y[i] = (x[p].key < x[q].key)? x[p++]: x[q++];
	}
	for (i = 0; i < n; i++) x[i] = y[i];
}

void add_BIT(int N, long long BIT[], int i, int k)
{
	while (i <= N) {
		BIT[i] += k;
		i += (i & -i);
	}
}

long long sum_BIT(long long BIT[], int r)
{
	long long sum = 0;
	while (r > 0) {
		sum += BIT[r];
		r -= (r & -r);
	}
	return sum;
}

int main()
{
	int i, N, A[200001];
	scanf("%d", &N);
	for (i = 1; i <= N; i++) scanf("%d", &(A[i]));
	
	int j, k, P[200001];
	data d[200001];
	for (i = 0; i < N; i++) {
		d[i].key = A[i+1];
		d[i].id = i + 1;
	}
	merge_sort(d, N);
	d[N].key = -1;
	for (i = 0, k = 1; i < N; i = j, k++) {
		for (j = i; d[j].key == d[i].key; j++) P[d[j].id] = k;
	}
	k--;
	
	long long BIT[2][200001] = {}, ans = 0;
	for (i = N; i >= 1; i--) {
		ans += A[i] * (sum_BIT(BIT[1], P[i] - 1) % Mod) % Mod;
		add_BIT(k, BIT[1], P[i], sum_BIT(BIT[0], P[i] - 1));
		add_BIT(k, BIT[0], P[i], 1);
	}
	for (i = 1; i <= k; i++) {
		BIT[0][i] = 0;
		BIT[1][i] = 0;
	}
	for (i = 1; i <= N; i++) {
		ans += A[i] * (sum_BIT(BIT[1], k - P[i]) % Mod) % Mod;
		add_BIT(k, BIT[1], k - P[i] + 1, sum_BIT(BIT[0], k - P[i]));
		add_BIT(k, BIT[0], k - P[i] + 1, 1);
	}
	for (i = 1; i <= k; i++) BIT[1][i] = 0;
	for (i = N; i >= 1; i--) {
		add_BIT(k, BIT[0], k - P[i] + 1, -1);
		ans += A[i] * (sum_BIT(BIT[0], k - P[i]) % Mod) % Mod * (sum_BIT(BIT[1], P[i] - 1) % Mod) % Mod;
		add_BIT(k, BIT[1], P[i], 1);
	}
	
	printf("%lld\n", ans % Mod);
	fflush(stdout);
	return 0;
}
0