#586012 (C) No.1300 Sum of Inversions

提出ソース

結果

問題	No.1300 Sum of Inversions
ユーザー	👑 ygussany
提出日時	2020-11-27 23:52:59
言語	C (gcc 12.3.0)
結果	AC
実行時間	101 ms / 2,000 ms
コード長	1,826 bytes
コンパイル時間	507 ms
コンパイル使用メモリ	31,976 KB
実行使用メモリ	9,728 KB
最終ジャッジ日時	2023-10-09 21:52:54
合計ジャッジ時間	4,716 ms
ジャッジサーバーID （参考情報）	judge12 / judge15

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	3 ms 7,076 KB
testcase_01	AC	2 ms 6,960 KB
testcase_02	AC	2 ms 6,472 KB
testcase_03	AC	79 ms 8,868 KB
testcase_04	AC	78 ms 9,100 KB
testcase_05	AC	62 ms 8,544 KB
testcase_06	AC	90 ms 9,384 KB
testcase_07	AC	85 ms 9,220 KB
testcase_08	AC	95 ms 9,436 KB
testcase_09	AC	93 ms 9,508 KB
testcase_10	AC	51 ms 8,152 KB
testcase_11	AC	51 ms 8,496 KB
testcase_12	AC	77 ms 8,928 KB
testcase_13	AC	75 ms 8,904 KB
testcase_14	AC	101 ms 9,620 KB
testcase_15	AC	92 ms 9,368 KB
testcase_16	AC	80 ms 9,088 KB
testcase_17	AC	50 ms 8,548 KB
testcase_18	AC	56 ms 8,196 KB
testcase_19	AC	68 ms 9,080 KB
testcase_20	AC	70 ms 8,792 KB
testcase_21	AC	69 ms 8,600 KB
testcase_22	AC	62 ms 8,636 KB
testcase_23	AC	90 ms 9,232 KB
testcase_24	AC	64 ms 8,896 KB
testcase_25	AC	54 ms 8,332 KB
testcase_26	AC	54 ms 8,464 KB
testcase_27	AC	59 ms 8,416 KB
testcase_28	AC	95 ms 9,388 KB
testcase_29	AC	68 ms 9,116 KB
testcase_30	AC	93 ms 9,432 KB
testcase_31	AC	62 ms 8,780 KB
testcase_32	AC	64 ms 8,500 KB
testcase_33	AC	31 ms 9,600 KB
testcase_34	AC	44 ms 9,644 KB
testcase_35	AC	65 ms 9,528 KB
testcase_36	AC	69 ms 9,728 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#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;
}

yukicoder

結果

テストケース

ソースコード