結果

問題 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
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

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