結果

問題 No.3153 probability max K
ユーザー Cafe1942
提出日時 2025-05-10 11:09:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 131 ms / 2,000 ms
コード長 1,027 bytes
コンパイル時間 757 ms
コンパイル使用メモリ 66,892 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-05-10 11:54:27
合計ジャッジ時間 5,751 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
using namespace std;
long long inv(long long base) {
    int i;
    long long res;
    long long pow2[30];
    pow2[0] = base;
    for (i = 1;i < 30;i++) {
        pow2[i] = ((long long)pow2[i - 1] * (long long)pow2[i - 1]) % (long long)998244353;
    }
    res = 1;
    for (int i = 0;i < 30;i++) {
        if ((998244351 >> i) % 2 == 1) {
            res *= (long long)pow2[i];
            res %= (long long)998244353;
        }
    }
    return res;
}
int main() {
    int N;
    long long K;
    cin >> N >> K;
    long long ansUpper = 1;
    long long ansLower = 1;
    while (N--) {
        long long A;
        cin >> A;
        if (A >= K) {
            ansUpper *= K;
            ansUpper %= 998244353;
            ansUpper *= inv(A);
            ansUpper %= 998244353;
            ansLower *= K - 1;
            ansLower %= 998244353;
            ansLower *= inv(A);
            ansLower %= 998244353;
        }
    }
    cout << (ansUpper - ansLower + 998244353) % 998244353 << "\n";
	return 0;
}
0