結果

問題 No.1960 Guruguru Permutation
コンテスト
ユーザー Seniorious
提出日時 2022-10-15 19:36:52
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 715 bytes
コンパイル時間 1,788 ms
コンパイル使用メモリ 193,592 KB
最終ジャッジ日時 2025-02-08 06:53:18
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:12:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   12 |   scanf("%d %d %d", &n, &a, &b);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

#define File(a) freopen(a ".in", "r", stdin), freopen(a ".out", "w", stdout)

const int N = 200005;
const int Mod = 998244353;

int iv[N];

int main() {
  int n, a, b;
  scanf("%d %d %d", &n, &a, &b);
  if (a + b > n) {
    int now = a + b - n;
    n -= now, a -= now, b -= now;
  }
  iv[1] = 1;
  for (int i = 2; i <= n; ++i) iv[i] = 1ll * (Mod - Mod / i) * iv[Mod % i] % Mod;
  int ans = 0;
  for (int i = 0, Ca = 1, Cb = 1; i <= std::min(a, b); ++i) {
    ans = (ans + 1ll * Ca * Cb) % Mod;
    Ca = 1ll * Ca * (a - i) % Mod * iv[i + 1] % Mod;
    Cb = 1ll * Cb * (b - i) % Mod;
  }
  for (int i = a + b + 1; i <= n; ++i) ans = 1ll * ans * i % Mod;
  printf("%d\n", ans);
  return 0;
}
0