結果

問題 No.1697 Deque House
ユーザー eSeFeSeF
提出日時 2021-09-17 23:52:40
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 115 ms / 3,500 ms
コード長 1,193 bytes
コンパイル時間 5,277 ms
コンパイル使用メモリ 264,936 KB
実行使用メモリ 33,988 KB
最終ジャッジ日時 2023-09-26 15:53:15
合計ジャッジ時間 8,242 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,384 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 1 ms
4,376 KB
testcase_07 AC 1 ms
4,376 KB
testcase_08 AC 2 ms
4,380 KB
testcase_09 AC 114 ms
33,916 KB
testcase_10 AC 68 ms
19,932 KB
testcase_11 AC 72 ms
20,124 KB
testcase_12 AC 56 ms
16,784 KB
testcase_13 AC 24 ms
8,732 KB
testcase_14 AC 39 ms
12,896 KB
testcase_15 AC 73 ms
19,892 KB
testcase_16 AC 74 ms
18,800 KB
testcase_17 AC 37 ms
12,344 KB
testcase_18 AC 46 ms
13,600 KB
testcase_19 AC 40 ms
13,604 KB
testcase_20 AC 113 ms
33,824 KB
testcase_21 AC 114 ms
33,988 KB
testcase_22 AC 115 ms
33,880 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