結果

問題 No.2356 Back Door Tour in Four Seasons
ユーザー lam6er
提出日時 2025-03-26 15:53:28
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 249 ms / 2,000 ms
コード長 1,284 bytes
コンパイル時間 363 ms
コンパイル使用メモリ 82,376 KB
実行使用メモリ 134,692 KB
最終ジャッジ日時 2025-03-26 15:54:05
合計ジャッジ時間 6,960 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

def main():
    import sys
    input = sys.stdin.read().split()
    idx = 0
    N = int(input[idx])
    idx += 1
    
    U = []
    F = []
    W = []
    K_P = 0
    A_all = 0
    
    for _ in range(N):
        S = input[idx]
        A = int(input[idx+1])
        idx += 2
        A_all += A
        if S == 'U':
            U.append(A)
        elif S == 'F':
            F.append(A)
        elif S == 'W':
            W.append(A)
        elif S == 'P':
            K_P += 1
    
    if N == 1:
        print(0)
        return
    
    numerator = (N - 2) % MOD
    denominator = (N - 1) % MOD
    inv_denominator = pow(denominator, MOD - 2, MOD)
    x = (numerator * inv_denominator) % MOD
    
    U_sum = 0
    for a in U:
        term = (1 - pow(x, a, MOD)) % MOD
        U_sum = (U_sum + term) % MOD
    
    F_sum = 0
    for a in F:
        term = (1 - pow(x, a, MOD)) % MOD
        F_sum = (F_sum + term) % MOD
    
    W_sum = 0
    for a in W:
        term = (1 - pow(x, a, MOD)) % MOD
        W_sum = (W_sum + term) % MOD
    
    part = U_sum * F_sum % MOD
    part = part * W_sum % MOD
    part = part * K_P % MOD
    
    pow_term = pow(denominator, A_all, MOD)
    ans = part * pow_term % MOD
    print(ans)
    
if __name__ == "__main__":
    main()
0