結果

問題 No.2874 Gunegune Tree
ユーザー hongrockhongrock
提出日時 2024-09-06 22:45:26
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 27 ms / 2,000 ms
コード長 1,188 bytes
コンパイル時間 2,168 ms
コンパイル使用メモリ 202,552 KB
実行使用メモリ 7,296 KB
最終ジャッジ日時 2024-09-06 22:45:30
合計ジャッジ時間 3,767 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 3 ms
6,940 KB
testcase_04 AC 12 ms
6,944 KB
testcase_05 AC 6 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 25 ms
7,168 KB
testcase_08 AC 14 ms
6,944 KB
testcase_09 AC 3 ms
6,944 KB
testcase_10 AC 22 ms
6,944 KB
testcase_11 AC 18 ms
6,940 KB
testcase_12 AC 7 ms
6,944 KB
testcase_13 AC 26 ms
7,040 KB
testcase_14 AC 21 ms
6,940 KB
testcase_15 AC 5 ms
6,940 KB
testcase_16 AC 26 ms
7,168 KB
testcase_17 AC 5 ms
6,944 KB
testcase_18 AC 5 ms
6,940 KB
testcase_19 AC 4 ms
6,940 KB
testcase_20 AC 5 ms
6,940 KB
testcase_21 AC 10 ms
6,940 KB
testcase_22 AC 13 ms
6,940 KB
testcase_23 AC 10 ms
6,944 KB
testcase_24 AC 12 ms
6,940 KB
testcase_25 AC 19 ms
6,944 KB
testcase_26 AC 13 ms
6,944 KB
testcase_27 AC 26 ms
7,040 KB
testcase_28 AC 8 ms
6,944 KB
testcase_29 AC 16 ms
6,940 KB
testcase_30 AC 13 ms
6,940 KB
testcase_31 AC 10 ms
6,940 KB
testcase_32 AC 27 ms
7,296 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

#define pb emplace_back
#define mp make_pair
 
using ll = long long;
using pii = pair<int,int>;
 
constexpr int mod = 998244353;
constexpr int inf = 0x3f3f3f3f;
constexpr int N = 2e5 + 10;

ll pow_mod(ll x, ll p){
  ll s = 1;
  while(p){
    if(p & 1) s = s * x % mod;
    x = x * x % mod;
    p >>= 1;
  }
  return s;
}

void add(int &x, int y){
  x += y;
  if(x >= mod)  x -= mod;
}

ll P[10];
int n, dp[N][5], cnt[5];

void _main(){
  int n;
  for(int i=2; i<=5; ++i) P[i] = pow_mod(i, mod - 2);
  for(int i=0; i<5; ++i){
    dp[1][i] = P[5];
  }
  for(int i=0; i<5; ++i){
    cnt[i] = 0;
    for(int j=0; j<5; ++j){
      if(abs(i - j) <= 1){
        ++cnt[i];
      }
    }
  }
  cin >> n;
  int ans = 1ll * 3 * P[5] % mod;
  for(int i=1; i<n; ++i){
    for(int j=0; j<5; ++j){
      for(int k=0; k<5; ++k){
        if(abs(j - k) <= 1){
          int tmp = 1ll * dp[i][k] * P[cnt[k]] % mod;
          add(dp[i+1][j], tmp);
          if(j > 0 && j < 4){
            add(ans, tmp);
          }
        }
      }
    }
  }
  cout << ans << '\n';
}

int main(){
  ios::sync_with_stdio(0);
  cin.tie(0); cout.tie(0);
  _main();
  return 0;
}
0