結果

問題 No.2356 Back Door Tour in Four Seasons
ユーザー AngrySadEightAngrySadEight
提出日時 2023-04-26 16:43:00
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 633 ms / 2,000 ms
コード長 1,581 bytes
コンパイル時間 771 ms
コンパイル使用メモリ 78,836 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-04-27 18:58:38
合計ジャッジ時間 14,285 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 623 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 1 ms
6,944 KB
testcase_08 AC 627 ms
6,940 KB
testcase_09 AC 629 ms
6,940 KB
testcase_10 AC 633 ms
6,940 KB
testcase_11 AC 630 ms
6,944 KB
testcase_12 AC 629 ms
6,944 KB
testcase_13 AC 630 ms
6,944 KB
testcase_14 AC 629 ms
6,940 KB
testcase_15 AC 313 ms
6,940 KB
testcase_16 AC 583 ms
6,940 KB
testcase_17 AC 602 ms
6,940 KB
testcase_18 AC 575 ms
6,940 KB
testcase_19 AC 319 ms
6,940 KB
testcase_20 AC 561 ms
6,940 KB
testcase_21 AC 460 ms
6,944 KB
testcase_22 AC 319 ms
6,944 KB
testcase_23 AC 230 ms
6,948 KB
testcase_24 AC 594 ms
6,940 KB
testcase_25 AC 17 ms
6,940 KB
testcase_26 AC 458 ms
6,944 KB
testcase_27 AC 592 ms
6,944 KB
testcase_28 AC 2 ms
6,940 KB
testcase_29 AC 2 ms
6,940 KB
testcase_30 AC 613 ms
6,940 KB
testcase_31 AC 616 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cassert>
using namespace std;
typedef long long ll;

ll mod2 = 998244353;

ll my_pow(ll x, ll n, ll mod){
    //　繰り返し二乗法．x^nをmodで割った余り．
    ll ret;
    if (n == 0){
        ret = 1;
    }
    else if (n % 2 == 1){
        ret = (x * my_pow((x * x) % mod, n / 2, mod)) % mod;
    }
    else{
        ret = my_pow((x * x) % mod, n / 2, mod);
    }
    return ret;
}

ll inv(ll x, ll mod){
    return my_pow(x, mod - 2, mod);
}

int main(){
    ll N;
    cin >> N;
    assert((2 <= N && N <= 300000));
    vector<char> S(N);
    vector<ll> A(N);
    ll all_sum = 0;
    ll spring = 0;
    for (ll i = 0; i < N; i++){
        cin >> S[i] >> A[i];
        assert((S[i] == 'U' || S[i] == 'F' || S[i] == 'W' || S[i] == 'P'));
        assert((1 <= A[i] && A[i] <= 100000));
        all_sum += A[i];
        if (S[i] == 'P') spring++;
    }
    ll summer = 0;
    ll fall = 0;
    ll winter = 0;
    for (ll i = 0; i < N; i++){
        ll prd = my_pow(N - 1, A[i], mod2);
        prd = (prd + mod2 - my_pow(N - 2, A[i], mod2)) % mod2;
        prd = (prd * inv(my_pow(N - 1, A[i], mod2) , mod2)) % mod2;
        if (S[i] == 'U') summer = (summer + prd) % mod2;
        if (S[i] == 'F') fall = (fall + prd) % mod2;
        if (S[i] == 'W') winter = (winter + prd) % mod2;
    }
    ll ans = 1;
    ans = (ans * summer) % mod2;
    ans = (ans * fall) % mod2;
    ans = (ans * winter) % mod2;
    ans = (ans * spring) % mod2;
    ans = (ans * my_pow(N - 1, all_sum, mod2)) % mod2;
    cout << ans << endl;
    
}
0