結果

問題 No.1696 Nonnil
ユーザー monnumonnu
提出日時 2021-10-01 22:54:22
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 153 ms / 3,500 ms
コード長 2,314 bytes
コンパイル時間 3,907 ms
コンパイル使用メモリ 233,728 KB
実行使用メモリ 40,528 KB
最終ジャッジ日時 2023-09-26 21:38:25
合計ジャッジ時間 8,690 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 44 ms
39,172 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 2 ms
4,380 KB
testcase_08 AC 47 ms
39,180 KB
testcase_09 AC 21 ms
5,960 KB
testcase_10 AC 31 ms
7,100 KB
testcase_11 AC 25 ms
5,464 KB
testcase_12 AC 26 ms
7,088 KB
testcase_13 AC 23 ms
5,684 KB
testcase_14 AC 121 ms
36,028 KB
testcase_15 AC 61 ms
8,576 KB
testcase_16 AC 146 ms
40,368 KB
testcase_17 AC 84 ms
32,384 KB
testcase_18 AC 86 ms
18,432 KB
testcase_19 AC 2 ms
4,380 KB
testcase_20 AC 2 ms
4,376 KB
testcase_21 AC 63 ms
26,092 KB
testcase_22 AC 89 ms
25,328 KB
testcase_23 AC 90 ms
23,364 KB
testcase_24 AC 86 ms
21,544 KB
testcase_25 AC 124 ms
31,984 KB
testcase_26 AC 93 ms
32,320 KB
testcase_27 AC 112 ms
39,288 KB
testcase_28 AC 123 ms
24,152 KB
testcase_29 AC 78 ms
28,688 KB
testcase_30 AC 63 ms
20,884 KB
testcase_31 AC 79 ms
34,220 KB
testcase_32 AC 90 ms
37,136 KB
testcase_33 AC 122 ms
27,412 KB
testcase_34 AC 65 ms
27,392 KB
testcase_35 AC 88 ms
30,480 KB
testcase_36 AC 148 ms
40,344 KB
testcase_37 AC 147 ms
40,300 KB
testcase_38 AC 148 ms
40,300 KB
testcase_39 AC 153 ms
40,528 KB
testcase_40 AC 147 ms
40,040 KB
testcase_41 AC 104 ms
11,216 KB
testcase_42 AC 40 ms
34,688 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll=long long;
using Graph=vector<vector<int>>;
#define MAX 20005
//#define MOD 1000000007
#define MOD 998244353
#define INF 1000000000
//#define INF 1000000000000000000

ll modpow(ll a,ll n,ll mod=MOD){
  ll res=1;
  while(n>0){
    if(n&1){
      res=(res*a)%mod;
    }
    a=(a*a)%mod;
    n>>=1;
  }
  return res;
}

ll modinv(ll a,ll mod=MOD){
  return modpow(a,mod-2,mod);
}

ll fac[MAX],finv[MAX],inv[MAX];

void COMinit(){
  fac[0]=fac[1]=1;
  finv[0]=finv[1]=1;
  inv[1]=1;
  for(int i=2;i<MAX;i++){
    fac[i]=(fac[i-1]*i)%MOD;
    inv[i]=MOD-inv[MOD%i]*(MOD/i)%MOD;
    finv[i]=finv[i-1]*inv[i]%MOD;
  }
}

ll COM(int n,int k){
  if(n<k){
    return 0;
  }
  if(n<0||k<0){
    return 0;
  }
  return fac[n]*(finv[k]*finv[n-k]%MOD)%MOD;
}



int main(){
  int N,K,M;
  cin>>N>>K>>M;
  vector<pair<int,int>> RL(M);
  for(int i=0;i<M;i++){
    cin>>RL[i].second>>RL[i].first;
  }
  sort(RL.begin(),RL.end());

  vector<vector<ll>> dp(K+1,vector<ll>(K+1,0));
  vector<vector<ll>> sum(K+1,vector<ll>(K+1,0));
  dp[0][0]=1;
  sum[0][0]=1;
  int k=0;
  int l=0;
  for(int i=1;i<=K;i++){
    while(k<M&&RL[k].first==i-1){
      l=max(l,RL[k].second);
      k++;
    }
    for(int j=0;j<K;j++){
      if(l==0){
        dp[i][j+1]+=sum[i-1][j];
        dp[i][j+1]%=MOD;
      }else{
        dp[i][j+1]+=sum[i-1][j]-sum[l-1][j]+MOD;
        dp[i][j+1]%=MOD;
      }
    }
    /*
    cout<<i<<" : ";
    for(int j=1;j<=K;j++){
      cout<<dp[i][j]<<' ';
    }
    cout<<'\n';
    */

    for(int j=0;j<=K;j++){
      sum[i][j]=sum[i-1][j]+dp[i][j];
      sum[i][j]%=MOD;
    }
  }
  while(k<M){
    l=max(l,RL[k].second);
    k++;
  }

  vector<ll> cnt(K+1,0);
  for(int i=l;i<=K;i++){
    for(int j=1;j<=min(K,N);j++){
      cnt[j]+=dp[i][j];
      cnt[j]%=MOD;
    }
  }

  vector<ll> pow_n(K+1,1);
  for(int i=1;i<=K;i++){
    pow_n[i]=modpow(i,N);
  }

  COMinit();

  ll ans=0;
  for(int i=1;i<=K;i++){
    //cout<<i<<" : "<<cnt[i]<<'\n';
    ll x=0;
    int op=1;
    for(int j=i;j>0;j--){
      if(op==1){
        x+=COM(i,j)*pow_n[j];
        x%=MOD;
      }else{
        x+=MOD-COM(i,j)*pow_n[j]%MOD;
        x%=MOD;
      }
      op*=-1;
    }
    ans+=x*cnt[i]%MOD;
    ans%=MOD;
  }
  cout<<ans<<'\n';
}
0