#790823 (C) No.2045 Two Reflections

• 問題: No.2045 Two Reflections
• ユーザー: 👑 ygussany
• 提出日時: 2022-08-20 14:20:50
• 言語: C (gcc 12.3.0)
• 結果: WA
• 実行時間: -
• コード長: 1,294 bytes
• コンパイル時間: 1,164 ms
• コンパイル使用メモリ: 31,488 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-09 07:09:48
• 合計ジャッジ時間: 1,286 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	3 ms; 5,248 KB
testcase_03	AC	3 ms; 5,248 KB
testcase_04	WA	-
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	2 ms; 5,248 KB
testcase_07	AC	3 ms; 5,248 KB
testcase_08	AC	3 ms; 5,248 KB
testcase_09	AC	3 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	WA	-
testcase_12	AC	1 ms; 5,248 KB
testcase_13	AC	1 ms; 5,248 KB
testcase_14	AC	1 ms; 5,248 KB
testcase_15	AC	1 ms; 5,248 KB
testcase_16	AC	2 ms; 5,248 KB
testcase_17	AC	1 ms; 5,248 KB
testcase_18	AC	1 ms; 5,248 KB
testcase_19	AC	1 ms; 5,248 KB
testcase_20	AC	3 ms; 5,248 KB
testcase_21	AC	3 ms; 5,248 KB
testcase_22	AC	3 ms; 5,248 KB
testcase_23	AC	4 ms; 5,248 KB
testcase_24	AC	3 ms; 5,248 KB
testcase_25	AC	4 ms; 5,248 KB
testcase_26	AC	3 ms; 5,248 KB
testcase_27	AC	4 ms; 5,248 KB
testcase_28	AC	3 ms; 5,248 KB
testcase_29	AC	3 ms; 5,248 KB

ソースコード

#include <stdio.h>

const int Mod = 998244353;

int main()
{
	int N, p, q;
	scanf("%d %d %d", &N, &p, &q);
	if (p == 1 || q == 1) {
		if (p == 1 && q == 1) printf("1\n");
		else printf("2\n");
		fflush(stdout);
		return 0;
	}
	
	int i, j, A[100001];
	for (i = 1; i <= N; i++) A[i] = i;
	for (i = 1, j = p; i < j; i++, j--) {
		A[i] ^= A[j];
		A[j] ^= A[i];
		A[i] ^= A[j];
	}
	for (i = N - q + 1, j = N; i < j; i++, j--) {
		A[i] ^= A[j];
		A[j] ^= A[i];
		A[i] ^= A[j];
	}
	
	int jj, n, k = 0, prime[100001], prime_inv[100001], mul[100001] = {}, flag[100001] = {};
	for (i = 2; i <= N; i++) {
		if (flag[i] != 0) continue;
		prime[k] = i;
		prime_inv[i] = k++;
		for (j = i * 2; j <= N; j += i) flag[j] = 1;
	}
	for (i = 1; i <= N; i++) flag[i] = 0;
	for (i = 1; i <= N; i++) {
		if (flag[i] != 0) continue;
		for (j = i, jj = A[i], flag[i] = 1; flag[jj] == 0; j = jj, jj = A[j]) flag[jj] = flag[j] + 1;
		if (flag[j] == 1) continue;
		for (n = flag[j], j = 0; j < k && prime[j] * prime[j] <= n; j++) {
			for (jj = 0; n % prime[j] == 0; n /= prime[j], jj++);
			if (mul[j] < jj) mul[j] = jj;
		}
		if (n > 1) mul[prime_inv[n]] = 1;
	}
	
	long long ans = 2;
	for (i = 0; i < k; i++) for (j = 0; j < mul[i]; j++) ans = ans * prime[i] % Mod;
	printf("%lld\n", ans);
	fflush(stdout);
	return 0;
}