結果

問題 No.1696 Nonnil
ユーザー platinumplatinum
提出日時 2020-11-20 12:48:59
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 558 ms / 3,500 ms
コード長 2,136 bytes
コンパイル時間 1,038 ms
コンパイル使用メモリ 87,024 KB
実行使用メモリ 134,884 KB
最終ジャッジ日時 2024-07-19 09:42:04
合計ジャッジ時間 11,518 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
7,620 KB
testcase_01 AC 2 ms
7,620 KB
testcase_02 AC 3 ms
7,624 KB
testcase_03 AC 114 ms
97,736 KB
testcase_04 AC 2 ms
7,496 KB
testcase_05 AC 3 ms
7,624 KB
testcase_06 AC 3 ms
7,624 KB
testcase_07 AC 2 ms
7,628 KB
testcase_08 AC 74 ms
59,972 KB
testcase_09 AC 39 ms
18,524 KB
testcase_10 AC 67 ms
23,656 KB
testcase_11 AC 43 ms
16,308 KB
testcase_12 AC 53 ms
22,568 KB
testcase_13 AC 40 ms
16,388 KB
testcase_14 AC 357 ms
120,152 KB
testcase_15 AC 115 ms
27,336 KB
testcase_16 AC 437 ms
134,884 KB
testcase_17 AC 275 ms
110,016 KB
testcase_18 AC 221 ms
62,368 KB
testcase_19 AC 3 ms
7,748 KB
testcase_20 AC 3 ms
7,632 KB
testcase_21 AC 208 ms
89,452 KB
testcase_22 AC 264 ms
86,068 KB
testcase_23 AC 246 ms
77,920 KB
testcase_24 AC 243 ms
73,316 KB
testcase_25 AC 364 ms
106,732 KB
testcase_26 AC 312 ms
110,064 KB
testcase_27 AC 379 ms
131,480 KB
testcase_28 AC 319 ms
80,908 KB
testcase_29 AC 265 ms
98,236 KB
testcase_30 AC 191 ms
71,416 KB
testcase_31 AC 274 ms
116,060 KB
testcase_32 AC 317 ms
125,932 KB
testcase_33 AC 340 ms
91,696 KB
testcase_34 AC 222 ms
92,964 KB
testcase_35 AC 282 ms
104,000 KB
testcase_36 AC 558 ms
134,556 KB
testcase_37 AC 553 ms
134,812 KB
testcase_38 AC 532 ms
134,684 KB
testcase_39 AC 404 ms
98,452 KB
testcase_40 AC 496 ms
116,112 KB
testcase_41 AC 206 ms
36,240 KB
testcase_42 AC 120 ms
104,392 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cassert>
#include <algorithm>
#define rep(i,n) for(int i=0; i<(int)(n); i++)

using namespace std;
using LL = long long;
using P = pair<int,int>;

const int Max_N = 1e9;
const int Max_K = 1500;
const int Max_M = 2e5;
const LL mod = 998244353;

LL modpow(LL x, LL n){
  LL r = 1;
  while(n){
    if(n & 1) r = r * x % mod;
    x = x * x % mod;
    n >>= 1;
  }
  return r;
}

LL dp[2000][2000][2];
LL sum1[2000][2000][2], sum2[4000][2000][2];

int main(){
  int N, K, M;
  cin >> N >> K >> M;
  assert(1 <= N and N <= Max_N);
  assert(1 <= K and K <= Max_K);
  assert(1 <= M and M <= min(Max_M, K * (K+1) / 2));
  vector<P> S(M);
  vector<int> L(1), R(1);
  vector<bool> erase(M);
  rep(i,M){
    int l, r;
    cin >> l >> r;
    assert(1 <= l and l <= r and r <= K);
    S[i] = P(l, r);
  }
  sort(S.begin(), S.end());
  rep(i,M-1) assert(S[i] != S[i+1]);
  rep(i,M-1) if(S[i+1].second <= S[i].second) erase[i] = true;
  rep(i,M-1){
    if(erase[i]) continue;
    for(int j=i+1; j<M; j++){
      if(S[j].second <= S[i].second){
        erase[i] = true;
        break;
      }
    }
  }
  rep(i,M) if(!erase[i]){
    L.push_back(S[i].first);
    R.push_back(S[i].second);
  }
  M = L.size();
  sum1[0][0][0] = 1, sum2[Max_K][0][0] = 1;
  rep(i,M-1){
    int l = L[i+1], r = R[i+1], d = r - l + 1;
    int pre_r = R[i];
    int s = *(lower_bound(R.begin(), R.end(), l) - 1);
    rep(j,K+1){
      rep(p,2){
        if(j >= d) dp[j][r][p] += sum1[j-d][s][p^1];
        dp[j][r][p] += sum2[j-r+Max_K][pre_r][p^1] - sum2[j-r+Max_K][s][p^1];
        dp[j][r][p] %= mod;
        (sum1[j][r][p] += sum1[j][pre_r][p] + dp[j][r][p]) %= mod;
        (sum2[j-r+Max_K][r][p] += sum2[j-r+Max_K][pre_r][p] + dp[j][r][p]) %= mod;
      }
    }
  }
  vector<LL> power(K+1);
  rep(i,K+1) power[i] = modpow(i, N);
  LL res = 0, ans = 0;
  rep(j,K+1){
    rep(k,K+1){
      rep(p,2) (dp[j][k][p] *= power[K-j]) %= mod;
    }
  }
  rep(j,K+1){
    rep(k,K+1) (res += dp[j][k][1] - dp[j][k][0]) %= mod;
  }
  ans = (modpow(K, N) - res) % mod;
  while(ans < 0) ans += mod;
  cout << ans << endl;

  return 0;
}
0