結果

問題 No.1866 Unfair Tournament
ユーザー cologne
提出日時 2022-03-04 22:33:29
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 735 ms / 3,000 ms
コード長 1,235 bytes
コンパイル時間 4,487 ms
コンパイル使用メモリ 287,140 KB
実行使用メモリ 13,668 KB
最終ジャッジ日時 2024-07-18 21:05:47
合計ジャッジ時間 10,335 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/modint>
#include <atcoder/convolution>
#include <bits/stdc++.h>
using namespace std;

using Mint = atcoder::modint998244353;
int main()
{
    int N, A, B;
    cin >> N >> A >> B;
    Mint ab = Mint(A) / Mint(B);
    // prob of 0 winning when # of 0 is idx.
    vector<Mint> base = {0, 1};

    auto conv = [&](vector<Mint> p)
    {
        int N = p.size() - 1;
        vector<Mint> pc(N + 1);
        for (int i = 0; i <= N; ++i)
            pc[i] = 1 - p[i];

        vector<Mint> fact(2 * N + 1);
        fact[0] = 1;
        for (int i = 1; i <= 2 * N; ++i)
            fact[i] = i * fact[i - 1];
        for (int i = 0; i <= N; ++i)
        {
            Mint C = fact[N] / fact[N - i] / fact[i];
            pc[i] *= C;
            p[i] *= C;
        }
        auto v1 = atcoder::convolution(p, p);
        auto v2 = atcoder::convolution(p, pc);
        for (int i = 0; i <= 2 * N; ++i)
        {
            Mint C = fact[2 * N] / fact[i] / fact[2 * N - i];
            v1[i] += 2 * ab * v2[i];
            v1[i] /= C;
        }
        return v1;
    };

    for (int i = 0; i < N; ++i)
        base = conv(base);
    for (int i = 0; i < (1 << N); ++i)
        cout << (base[i + 1] - base[i]).val() << endl;
}
0