結果
問題 |
No.1887 K Consecutive Ks (Easy)
|
ユーザー |
![]() |
提出日時 | 2022-03-25 21:51:17 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 123 ms / 2,000 ms |
コード長 | 684 bytes |
コンパイル時間 | 4,831 ms |
コンパイル使用メモリ | 255,612 KB |
最終ジャッジ日時 | 2025-01-28 11:59:37 |
ジャッジサーバーID (参考情報) |
judge4 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for(int i=0;i<(n);i++) #define Inf 1000000001 int main(){ int N,M; cin>>N>>M; vector dp(N+1,vector<mint>(M+1,0)); dp[0][0] = 1; vector<mint> sum(N+1,0); sum[0] = 1; for(int i=1;i<=N;i++){ for(int j=1;j<=M;j++){ mint v = sum[i-1]; if(i-j>=0)v -= sum[i-j]; v -= dp[i-1][j]; if(i-j>=0)v += dp[i-j][j]; dp[i][j] = v; dp[i][j] += dp[i-1][j]; sum[i] += v; } sum[i] += sum[i-1]; } mint ans = sum.back(); ans -= sum[N-1]; ans = mint(M).pow(N) - ans; cout<<ans.val()<<endl; return 0; }