#731396 (C) No.1838 Modulo Straight

• 問題: No.1838 Modulo Straight
• ユーザー: 👑 ygussany
• 提出日時: 2022-01-16 15:40:15
• 言語: C (gcc 12.3.0)
• 結果: AC
• 実行時間: 192 ms / 2,000 ms
• コード長: 1,050 bytes
• コンパイル時間: 390 ms
• コンパイル使用メモリ: 31,360 KB
• 実行使用メモリ: 23,680 KB
• 最終ジャッジ日時: 2024-11-23 11:05:01
• 合計ジャッジ時間: 6,774 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	3 ms; 5,248 KB
testcase_01	AC	3 ms; 5,248 KB
testcase_02	AC	3 ms; 5,248 KB
testcase_03	AC	4 ms; 5,248 KB
testcase_04	AC	4 ms; 5,248 KB
testcase_05	AC	4 ms; 5,248 KB
testcase_06	AC	4 ms; 5,248 KB
testcase_07	AC	4 ms; 5,248 KB
testcase_08	AC	3 ms; 5,248 KB
testcase_09	AC	4 ms; 5,248 KB
testcase_10	AC	4 ms; 5,248 KB
testcase_11	AC	186 ms; 23,552 KB
testcase_12	AC	191 ms; 23,552 KB
testcase_13	AC	191 ms; 23,680 KB
testcase_14	AC	192 ms; 23,680 KB
testcase_15	AC	189 ms; 23,680 KB
testcase_16	AC	168 ms; 15,744 KB
testcase_17	AC	165 ms; 15,872 KB
testcase_18	AC	169 ms; 15,744 KB
testcase_19	AC	149 ms; 13,312 KB
testcase_20	AC	151 ms; 13,184 KB
testcase_21	AC	143 ms; 13,312 KB
testcase_22	AC	66 ms; 9,728 KB
testcase_23	AC	70 ms; 9,728 KB
testcase_24	AC	74 ms; 9,600 KB
testcase_25	AC	77 ms; 9,600 KB
testcase_26	AC	80 ms; 9,600 KB
testcase_27	AC	82 ms; 9,728 KB
testcase_28	AC	83 ms; 9,728 KB
testcase_29	AC	85 ms; 9,728 KB
testcase_30	AC	89 ms; 9,728 KB
testcase_31	AC	96 ms; 9,984 KB
testcase_32	AC	118 ms; 11,392 KB
testcase_33	AC	163 ms; 15,872 KB
testcase_34	AC	162 ms; 15,872 KB
testcase_35	AC	133 ms; 12,032 KB
testcase_36	AC	112 ms; 10,880 KB
testcase_37	AC	92 ms; 10,112 KB
testcase_38	AC	84 ms; 9,600 KB
testcase_39	AC	28 ms; 10,624 KB
testcase_40	AC	4 ms; 5,248 KB

ソースコード

#include <stdio.h>
#include <stdlib.h>

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

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

void chmin(long long* a, long long b)
{
	if (*a > b) *a = b;
}

int main()
{
	int i, M, K, N, A[400001], num[400001] = {}, *p[400001];
	scanf("%d %d", &M, &K);
	for (i = 0; i < M; i++) p[i] = (int*)malloc(sizeof(int) * (K + 1));
	for (i = 1, N = M * K; i <= N; i++) {
		scanf("%d", &(A[i]));
		p[A[i]][++num[A[i]]] = i;
		A[i] += M * (num[A[i]] - 1) + 1;
	}
	
	int j, BIT[400001] = {}, dif[400001];
	long long ans = 0, tmp;
	for (i = 1; i <= N; i++) {
		ans += i - sum_BIT(BIT, A[i]) - 1;
		dif[i] = (sum_BIT(BIT, ((A[i] - 1) / M + 1) * M) - sum_BIT(BIT, ((A[i] - 1) / M) * M)) * 2 - (M - 1);
		add_BIT(N, BIT, A[i], 1);
	}
	for (i = M - 1, tmp = ans; i >= 1; i--) {
		for (j = 1; j <= K; j++) tmp += dif[p[i][j]];
		chmin(&ans, tmp);
	}
	printf("%lld\n", ans);
	fflush(stdout);
	return 0;
}