結果

問題 No.3391 Line up Dominoes
コンテスト
ユーザー Rumain831
提出日時 2026-10-16 14:19:44
言語 C++23(gcc16)
(gcc 16.2.0 + boost 1.92.0 + ACL)
コンパイル:
g++-16 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 338 ms / 3,000 ms
+ 348µs
コード長 2,273 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,845 ms
コンパイル使用メモリ 188,576 KB
実行使用メモリ 9,924 KB
最終ジャッジ日時 2026-10-16 14:20:56
合計ジャッジ時間 12,753 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
using ll = long long;

const ll MOD=998244353;

struct mint{
  ll val;
  mint(ll val_=0):val(val_%MOD){
    if(val<0) val+=MOD; 
  }

  mint operator-()const{
    return mint(-val);
  }
  mint& operator+=(const mint& other){
    val+=other.val;
    if(val>=MOD) val-=MOD;
    return *this;
  }
  mint& operator-=(const mint& other){
    val-=other.val;
    if(val<0) val+=MOD;
    return *this;
  }
  mint& operator*=(const mint& other){
    val*=other.val, val%=MOD;
    return *this;
  }
  mint pow(ll n) const{
    mint ans(1);
    mint mul(*this);
    while(n){
      if(n&1) ans*=mul;
      mul*=mul;
      n/=2;
    }
    return ans;
  }
  mint inv() const{
    return pow(MOD-2);
  }
  mint& operator/=(const mint& other){
    return *this*=other.inv();
  }

  friend bool operator==(const mint& lhs, const mint& rhs){
    return lhs.val == rhs.val;
  }
  friend bool operator!=(const mint& lhs, const mint& rhs){
    return lhs.val != rhs.val;
  }
  friend mint operator+(mint lhs, const mint& rhs) {
    lhs+=rhs;
    return lhs;
  }
  friend mint operator-(mint lhs, const mint& rhs) {
    lhs-=rhs;
    return lhs;
  }
  friend mint operator*(mint lhs, const mint& rhs) {
    lhs*=rhs;
    return lhs;
  }
  friend mint operator/(mint lhs, const mint& rhs) {
    lhs/=rhs;
    return lhs;
  }
  friend istream& operator>>(istream& is, mint& m) {
    ll x;
    is >> x;
    m=mint(x);
    return is;
  }
  friend ostream& operator<<(ostream& os, const mint& m) {
    return os << m.val;
  }
};

int main(void){
  int n, m, k; cin >> n >> m >> k;
  vector<int> a(n);
  for(auto&x:a) cin >> x;
  sort(begin(a), end(a));
  vector<int> l(n), r(n);
  for(int i=0; i<n; i++){
    l[i]=lower_bound(begin(a), end(a), a[i]-k)-begin(a);
    r[i]=upper_bound(begin(a), end(a), a[i]+k)-begin(a);
  }
  vector<mint> dp(n), sum(n+1);
  for(int i=0; i<n; i++) dp[i]=1, sum[i+1]=sum[i]+dp[i];
  for(int i=1; i<m; i++){
    vector<mint> old(n), oldsum(n+1);
    swap(old, dp);
    swap(oldsum, sum);
    for(int j=0; j<n; j++) dp[j]=oldsum[r[j]]-oldsum[l[j]];
    for(int j=0; j<n; j++) sum[j+1]=sum[j]+dp[j];
  } 
  mint ans=0;
  for(int i=0; i<n; i++) ans+=dp[i];
  cout << ans << endl;
  return 0; 
}
0