結果

問題 No.1887 K Consecutive Ks (Easy)
コンテスト
ユーザー miscalc
提出日時 2022-03-07 15:15:53
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
TLE  
実行時間 -
コード長 1,035 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,448 ms
コンパイル使用メモリ 208,636 KB
最終ジャッジ日時 2025-01-28 07:43:01
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 8 TLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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

#include <atcoder/modint>
using namespace atcoder;
using mint = modint998244353;

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

int main()
{
  int N, M;
  cin >> N >> M;

  vector dp(2, vector(M + 1, vector(N + 1, mint(0))));
  dp[0][0][0] = 1;
  for (int i = 0; i < N; i++)
  {
    int pi = i & 1, ni = 1 - pi;
    for (int j = 0; j <= M; j++)
    {
      dp[ni][j].assign(N + 1, 0);
    }

    mint s = 0;
    for (int j = 0; j <= M; j++)
    {
      for (int k = 0; k <= N; k++)
      {
        if (k + 1 <= N && k + 1 < j)
          dp[ni][j][k + 1] += dp[pi][j][k];

        s += dp[pi][j][k];
        if (1 < j)
          dp[ni][j][1] -= dp[pi][j][k];
      }
    }
    for (int j = 2; j <= M; j++)
    {
      dp[ni][j][1] += s;
    }
  }

  mint ans = mint(M).pow(N);
  for (int j = 0; j <= M; j++)
  {
    for (int k = 0; k <= N; k++)
    {
      mint tmp = dp[N & 1][j][k];
      ans -= tmp;
    }
  }
  cout << ans.val() << endl;
}
0