結果
問題 | No.1431 東大文系数学2020第2問改 |
ユーザー |
👑 |
提出日時 | 2021-03-14 14:23:22 |
言語 | C (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 731 ms / 5,000 ms |
コード長 | 1,129 bytes |
コンパイル時間 | 540 ms |
コンパイル使用メモリ | 31,872 KB |
実行使用メモリ | 142,484 KB |
最終ジャッジ日時 | 2024-11-06 04:08:44 |
合計ジャッジ時間 | 8,705 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 28 |
ソースコード
#include <stdio.h>const int Mod = 998244353;long long fact[9000001], fact_inv[9000001];long long div_mod(long long x, long long y, long long z){if (x % y == 0) return x / y;else return (div_mod((1 + x / y) * y - x, (z % y), y) * z + x) / y;}long long combination(int n, int k){return fact[n] * fact_inv[k] % Mod * fact_inv[n-k] % Mod;}int main(){int N, M, K;scanf("%d %d %d", &N, &M, &K);int i;for (i = 1, fact[0] = 1; i <= N * N; i++) fact[i] = fact[i-1] * i % Mod;for (i = N * N - 1, fact_inv[N*N] = div_mod(1, fact[N*N], Mod); i >= 0; i--) fact_inv[i] = fact_inv[i+1] * (i + 1) % Mod;int j, k, x, y;long long ans = 0, tmp;for (x = 1; x <= N; x++) {for (y = 1; y <= N; y++) {if (K + x + y > N * 2 || x * y < M) continue;if ((N * 2 - K - x - y) % 2 == 0) ans += combination(x * y, M) * combination(N, x) % Mod * combination(N, y) % Mod * combination(N * 2 -(x + y), K) % Mod;else ans += Mod - combination(x * y, M) * combination(N, x) % Mod * combination(N, y) % Mod * combination(N * 2 - (x + y), K) % Mod;}}printf("%lld\n", ans % Mod);fflush(stdout);return 0;}