結果

問題 No.1696 Nonnil
ユーザー eSeF
提出日時 2021-09-15 15:47:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 564 ms / 3,500 ms
コード長 1,414 bytes
コンパイル時間 4,174 ms
コンパイル使用メモリ 257,440 KB
最終ジャッジ日時 2025-01-24 14:05:09
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

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;
  int M;
  cin >> M;
  vector<int>R(K + 1, -1);
  for(int i = 0; i < M; i++){
    int l, r;
    cin >> l >> r;
    R[r] = max(R[r], l);
  }

  for(int i = 0; i <= K; i++){
    if(R[i] == -1) continue;
    for(int j = i + 1; j <= K; j++){
      if(R[j] <= R[i]){
        R[j] = -1;
      }
    }
  }
  vector<vector<vector<mint>>>dp1(K + 1, vector<vector<mint>>(K + 1, vector<mint>(2, 0))), dp2(K + 1, vector<vector<mint>>(K + 1, vector<mint>(2, 0)));
  dp1[0][0][0] = dp2[0][0][0] = 1;
  for(int i = 1; i <= K; i++){
    for(int j = 0; j <= K; j++){
      vector<mint>v(2, 0);
      if(R[i] != -1 && j > 0){
        int L = R[i];
        for(int k = 0; k < 2; k++){
          v[k] += dp2[i - 1][j - 1][k ^ 1];
          if(j - (i - L + 1) >= 0){
            v[k] -= dp2[L - 1][j - (i - L + 1)][k ^ 1];
            v[k] += dp1[L - 1][j - (i - L + 1)][k ^ 1];
          }
        }
      }

      for(int k = 0; k < 2; k++){
        dp1[i][j][k] = v[k] + dp1[i - 1][j][k];

        dp2[i][j][k] = v[k];
        if(i - 1 >= 0 && j - 1 >= 0) dp2[i][j][k] += dp2[i - 1][j - 1][k];
      }
    }
  }
  mint ans = 0;
  for(int i = 0; i < K; i++){
    ans += (dp1[K][i][0] - dp1[K][i][1]) * mint(K - i).pow(N);
  }
  cout << ans.val() << endl;
  return 0;
}
0