結果
問題 | No.2874 Gunegune Tree |
ユーザー |
|
提出日時 | 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 |
ソースコード
#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;}