結果

問題 No.1887 K Consecutive Ks (Easy)
ユーザー miscalcmiscalc
提出日時 2022-03-07 15:15:53
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,035 bytes
コンパイル時間 2,584 ms
コンパイル使用メモリ 213,280 KB
実行使用メモリ 114,560 KB
最終ジャッジ日時 2024-07-22 10:02:13
合計ジャッジ時間 6,132 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
10,752 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 TLE -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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