結果

問題 No.1697 Deque House
ユーザー eSeFeSeF
提出日時 2021-09-17 23:52:40
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 104 ms / 3,500 ms
コード長 1,193 bytes
コンパイル時間 3,910 ms
コンパイル使用メモリ 268,200 KB
実行使用メモリ 34,176 KB
最終ジャッジ日時 2024-07-19 09:57:58
合計ジャッジ時間 6,438 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 104 ms
34,176 KB
testcase_10 AC 65 ms
19,968 KB
testcase_11 AC 64 ms
19,956 KB
testcase_12 AC 51 ms
16,896 KB
testcase_13 AC 22 ms
8,832 KB
testcase_14 AC 36 ms
13,056 KB
testcase_15 AC 66 ms
20,224 KB
testcase_16 AC 67 ms
18,688 KB
testcase_17 AC 33 ms
12,416 KB
testcase_18 AC 43 ms
13,824 KB
testcase_19 AC 37 ms
13,696 KB
testcase_20 AC 104 ms
34,048 KB
testcase_21 AC 104 ms
34,048 KB
testcase_22 AC 104 ms
34,048 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
int main(){
  int N, K;
  cin >> N >> K;
  vector<int>a(N);
  for(int i=0;i<N;i++)cin >> a[i];
  vector<vector<mint>>F(N+1,vector<mint>(K+1,0));
  for(int i=0;i<=K;i++)F[0][i] = 1;
  for(int i=0;i<N;i++){
      for(int j=0;j<=K;j++){
          F[i+1][j]+=F[i][j]*(1<<j);
      }
      for(int j=1;j<=K;j++)F[i+1][j]+=F[i+1][j-1];
  }

  vector<vector<mint>>dp(N+1,vector<mint>(K+1,0));
  vector<vector<mint>>rdp(N+1,vector<mint>(K+1,0));

  for(int i=0;i<N;i++){
    mint ai = 1, rai = 1;
    for(int j=0;j<=K;j++){
      dp[i+1][j]+=(dp[i][j]+F[i][j]*ai)*(1<<j);
      rdp[i+1][j]+=(rdp[i][j]+F[i][j]*rai)*(1<<j);

      ai *= a[i];
      rai *= a[N-1-i];
    }

    for(int j=1;j<=K;j++){
      dp[i+1][j]+=dp[i+1][j-1];
      rdp[i+1][j]+=rdp[i+1][j-1];
    }
  }

  mint ans = 0;
  for(int i=2;i<=N;i++){
    ans += dp[i-2][K]*(1<<(2*K))*F[N-i][K-1];
    ans += F[i-2][K]*(1<<(2*K))*F[N-i][K-1]*(mint(a[i-1]).pow(K)+mint(a[i-2]).pow(K));
    ans += F[i-2][K]*(1<<(2*K))*rdp[N-i][K-1];
  }
  ans *= mint(4).pow(K).inv();
  cout << ans.val() << endl;



  return 0;
}
0