結果

問題 No.2688 Cell Proliferation (Hard)
ユーザー 👑 rin204rin204
提出日時 2024-03-03 13:53:17
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 411 ms / 4,000 ms
コード長 1,619 bytes
コンパイル時間 4,511 ms
コンパイル使用メモリ 290,664 KB
実行使用メモリ 43,012 KB
最終ジャッジ日時 2024-09-30 06:44:37
合計ジャッジ時間 11,054 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,820 KB
testcase_03 AC 49 ms
8,064 KB
testcase_04 AC 411 ms
42,072 KB
testcase_05 AC 203 ms
23,420 KB
testcase_06 AC 103 ms
13,276 KB
testcase_07 AC 106 ms
13,100 KB
testcase_08 AC 400 ms
41,636 KB
testcase_09 AC 405 ms
41,168 KB
testcase_10 AC 404 ms
42,584 KB
testcase_11 AC 403 ms
41,960 KB
testcase_12 AC 98 ms
13,612 KB
testcase_13 AC 402 ms
43,012 KB
testcase_14 AC 404 ms
40,576 KB
testcase_15 AC 403 ms
43,012 KB
testcase_16 AC 403 ms
40,432 KB
testcase_17 AC 196 ms
23,720 KB
testcase_18 AC 406 ms
42,552 KB
testcase_19 AC 199 ms
23,632 KB
testcase_20 AC 193 ms
21,996 KB
testcase_21 AC 194 ms
21,984 KB
testcase_22 AC 197 ms
23,580 KB
testcase_23 AC 5 ms
6,816 KB
testcase_24 AC 9 ms
6,816 KB
testcase_25 AC 96 ms
13,532 KB
testcase_26 AC 405 ms
42,792 KB
testcase_27 AC 7 ms
6,816 KB
testcase_28 AC 7 ms
6,816 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#include "atcoder/convolution.hpp"
#include "atcoder/modint.hpp"
using mint = atcoder::modint998244353;

vector<mint> inv(vector<mint> &F) {
    int n          = F.size();
    vector<mint> G = {mint(1) / F[0]};
    int l          = 1;
    while (l < n) {
        vector<mint> tmp = G;
        for (int i = 0; i < l; i++) {
            tmp[i] *= 2;
        }
        l <<= 1;
        G = atcoder::convolution(G, G);
        vector<mint> tmp2;
        if (n >= l) {
            tmp2 = atcoder::convolution(vector<mint>({F.begin(), F.begin() + l}), G);
        } else {
            tmp2 = atcoder::convolution(F, G);
        }
        G.assign(l, 0);
        for (int i = 0; i < l; i++) {
            if (i < tmp.size()) G[i] += tmp[i];
            if (i < tmp2.size()) G[i] -= tmp2[i];
        }
    }
    G.resize(n);
    return G;
}

int main() {
    long long p1, p2, q1, q2, T;
    cin >> p1 >> p2 >> q1 >> q2 >> T;
    mint p = mint(p1) / p2;
    mint q = mint(q1) / q2;
    vector<mint> powQ(T + 1, 1);
    for (int i = 1; i <= T; i++) {
        powQ[i] = powQ[i - 1] * q;
    }

    if (p == 0) {
        mint ans = 1;
        for (int i = 1; i <= T; i++) {
            ans *= powQ[i];
        }
        cout << ans.val() << endl;
        return 0;
    }

    vector<mint> F(T + 2, 0);
    mint alive = 1;
    for (int i = 1; i < T + 2; i++) {
        F[i] = alive * p;
        alive *= powQ[i];
    }
    F[0] = 1;
    for (int i = 1; i < T + 2; i++) {
        F[i] = -F[i];
    }
    auto G   = inv(F);
    mint ans = G[T + 1] / p;
    cout << ans.val() << endl;
}
0