結果

問題 No.1771 A DELETEQ
コンテスト
ユーザー hitonanode
提出日時 2021-11-23 11:01:01
言語 C++17(gcc12)
(gcc 12.4.0 + boost 1.90.0)
コンパイル:
g++-12 -O2 -lm -std=c++17 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 38 ms / 3,500 ms
コード長 1,026 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 5,148 ms
コンパイル使用メモリ 264,864 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2026-06-24 10:01:19
合計ジャッジ時間 6,837 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 RE * 12
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

// Validator
#include "testlib.h"
#include <iostream>
#include <utility>
#include <vector>
using namespace std;

#include <atcoder/modint>
using mint = atcoder::modint998244353;

int main(int argc, char *argv[]) {
    registerValidation(argc, argv);
    int X = inf.readInt(1, 4000);
    inf.readSpace();
    int Y = inf.readInt(1, 4000);
    inf.readEoln();
    inf.readEof();

    if (X > Y) swap(X, Y);

    int K = Y - X;

    mint ret = 0;

    vector<mint> fac(X + Y + 1, 1), facinv = fac;
    for (int i = 1; i < int(fac.size()); ++i) fac[i] = fac[i - 1] * i;
    facinv.back() = fac.back().inv();
    for (int i = int(facinv.size()) - 1; i; --i) facinv[i - 1] = facinv[i] * i;

    mint pow2 = 1;  // 2^i
    for (int i = 0; i <= X; ++i) { // i: (# of AA) + (# of BB)
        for (int j = 0; i + j <= X; ++j) {  // j: (# of AB)
            int k = j + K; // k: (# of BA)
            ret += fac[i + j + k] * facinv[i] * facinv[j] * facinv[k] * pow2;
        }
        pow2 *= 2;
    }

    cout << ret.val() << '\n';
}
0