結果

問題 No.1771 A DELETEQ
コンテスト
ユーザー hitonanode
提出日時 2021-11-23 11:01:01
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,026 bytes
コンパイル時間 162 ms
コンパイル使用メモリ 24,704 KB
最終ジャッジ日時 2025-01-26 00:35:08
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
main.cpp:2:10: fatal error: testlib.h: No such file or directory
    2 | #include "testlib.h"
      |          ^~~~~~~~~~~
compilation terminated.

ソースコード

diff # 

// 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