結果

問題 No.2874 Gunegune Tree
コンテスト
ユーザー karinohito
提出日時 2024-09-06 22:18:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 10 ms / 2,000 ms
コード長 782 bytes
コンパイル時間 2,007 ms
コンパイル使用メモリ 199,356 KB
最終ジャッジ日時 2025-02-24 04:29:47
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<atcoder/modint>
using namespace atcoder;
using mint=modint998244353;
using namespace std;
using ll=long long;


int main() {
    ll N;
    cin>>N;
    vector<mint> P(5);
    vector<mint> iv(6,0);
    for(int i=1;i<6;i++)iv[i]=mint(i).inv();
    for(int i=0;i<5;i++)P[i]=iv[5];
    mint an=0;
    for(int i=0;i<N-1;i++){
        vector<mint> NP(5);
        for(int i=1;i<4;i++)an+=P[i];
        for(int i=0;i<5;i++){
            if(i==0||i==4)NP[i]=P[0]*iv[2]+P[1]*iv[3];
            else if(i==2){
                NP[i]=P[2]*iv[3]+(P[1]+P[3])*iv[3];
            }
            else{
                NP[i]=P[2]*iv[3]+P[1]*iv[3]+P[0]*iv[2];
            }
        }
        swap(P,NP);
    }
    for(int i=1;i<4;i++)an+=P[i];
    cout<<an.val()<<endl;
}
0