#1163966 (C++23) No.3538 Not First Place

• 問題: No.3538 Not First Place
• ユーザー: ZeriToki
• 提出日時: 2026-05-08 23:09:57
• 言語: C++23 (gcc 15.2.0 + boost 1.89.0)
• 結果: AC
• 実行時間: 804 ms / 5,000 ms
• コード長: 3,835 bytes
• コンパイル時間: 2,246 ms
• コンパイル使用メモリ: 344,884 KB
• 実行使用メモリ: 495,680 KB
• 最終ジャッジ日時: 2026-05-08 23:10:10
• 合計ジャッジ時間: 9,055 ms
• ジャッジサーバーID （参考情報）: judge2_0 / judge1_0

ファイルパターン	結果
sample	AC * 2
other	AC * 26

ソースコード

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
template<typename T> bool chmin(T& a, T b){if(a > b){a = b; return true;} return false;}
template<typename T> bool chmax(T& a, T b){if(a < b){a = b; return true;} return false;}
const long long mod=998244353;
const long long mod2=469762049;
struct MATH{
    vector<int>factorial;//階乗データ
    vector<int>factorial_inv;
    void factorial_init(ll N,ll m){//階乗を入れる
        assert(N>=0);
        factorial.assign(N+1,0);
        factorial_inv.assign(N+1,0);
        ll p=1;
        factorial[0]=1;
        for(int i=1;i<=N;i++){
            p=(p*i)%m;
            factorial[i]=p;
        }
        factorial_inv[N]=modinv(factorial[N],m);
        p=factorial_inv[N];
        for(int i=N-1;i>=0;i--){
            p=p*(i+1)%m;
            factorial_inv[i]=p;
        }
        return;
    }
    ll power(ll a,ll b,ll m){//a^b
        assert(b>=0);
        a%=m;
        if(a<0) a+=m;
        ll p=a,Answer=1;
        for(ll i=0;i<60;i++){
            ll wari=(1LL<<i);
            if((b/wari)%2==1){
                Answer=(Answer*p)%m;
            }
            p=(p*p)%m;
        }
        return Answer;
    }
    ll Division(ll a,ll b,ll m){//a/b
        return (a*power(b,m-2,m))%m;
    }
    ll modinv(ll x,ll m){//逆元
        return power(x,m-2,m);
    }
    ll comb(ll n,ll r,ll m){//nCrを求める(factorial必要)
        if(n<0 || r<0 || n-r<0) return 0;
        ll res=factorial[n];
        res=(res*factorial_inv[r])%m;
        res=(res*factorial_inv[n-r])%m;
        return res;
    }
    ll GCD(ll A,ll B){//GCD(A,B)
        while(A>=1 && B>=1){
            if(A>=B) A=(A%B);
            else B=(B%A);
        }
        if(A!=0) return A;
        return B;
    }
    ll modlog(ll a,ll b,ll m){//a^x=b(mod m)を満たす最小のx(ないなら-1)
        a%=m;b%=m;
        ll lo=-1,hi=m;
        while(hi-lo>1){
            ll mid=(lo+hi)/2;
            if(mid*mid>=m) hi=mid;
            else lo=mid;
        }
        ll sqrtM=hi;
        map<ll,ll>baby;
        ll amari=a;
        for(ll r=1;r<sqrtM;r++){
            if(!baby.count(amari)) baby[amari]=r;
            amari=(amari*a)%m;
        }
        ll A=power(modinv(a,m),sqrtM,m);
        amari=b;
        for(ll q=0;q<sqrtM;q++){
            if(amari==1 && q>=1) return q*sqrtM;
            else if(baby.count(amari)) return q*sqrtM+baby[amari];
            amari=(amari*A)%m;
        }
        return -1;
    }
};
MATH math;
vector<ll>conv(vector<ll>&a,vector<ll>&b){
    vector<ll>res;
    res.assign(a.size()+b.size(),0);
    for(int i=0;i<(int)a.size();i++){
        if(a[i]==0) continue;
        for(int j=0;j<(int)b.size();j++){
            if(b[j]==0) continue;
            res[i+j]=(res[i+j]+a[i]*b[j])%mod;
        }
    }
    return res;
}
int main(){
    cout.tie()->sync_with_stdio(0);
    cin.tie(0);
    ll N,M,K,L;cin>>N>>M>>K>>L;
    math.factorial_init((N+M*K),mod);
    ll ans=0;
    vector<ll>a,b;
    a.assign(M+1,0);
    b.assign(M*N*3,0);
    for(int i=L;i<=M;i++) a[i]=1;
    for(int i=0;i<N;i++){
        b[i*(M+1)]=math.comb(N-1,i,mod);
        if(i%2==1){
            b[i*(M+1)]*=-1;
            if(b[i*(M+1)]<0) b[i*(M+1)]+=mod;
        }
    }
    vector<ll>c=conv(b,a);
    int k;
    ll plus;
    for(int i=0;i<(int)c.size();i++){
        if(i>M*K) break;
        k=M*K-i;
        plus=c[i]*math.comb(N+k-2,N-2,mod)%mod;
        ans=(ans+plus)%mod;
    }
    for(int i=L;i<=M;i++){
        for(int j=0;j<N;j++){
            k=M*K-i-j*(i+1);
            if(k<0) continue;
            plus=math.comb(N-1,j,mod);
            if(j%2==1) plus*=-1;
            plus=plus*math.comb(N+k-2,N-2,mod)%mod;
            ans=(ans-plus)%mod;
        }
    }
    if(ans<0) ans+=mod;
    cout<<ans<<endl;
}