結果

問題 No.1887 K Consecutive Ks (Easy)
ユーザー miscalc
提出日時 2022-03-07 15:15:53
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,035 bytes
コンパイル時間 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 #
プレゼンテーションモードにする

#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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0