萌新求助,过不了样例
查看原帖
萌新求助,过不了样例
610557
shinzanmonoszm 妹妹楼主2023/8/8 12:35
#include<iostream>
#include<algorithm>
using ll=long long;
const int sz=2010;
const int mod=998244353;
ll f[2][sz][sz],sum[2][sz][sz],fact[sz],inv[sz];
int bl[sz],br[sz];
ll qpow(ll base,ll exp){
    ll ans=1;
    while(exp!=0){
        if(exp&1)ans=ans*base%mod;
        base=base*base%mod,exp>>=1;
    }
    return ans;
}
int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int n,K;
    std::cin>>n>>K;
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=fact[i-1]*i%mod;
    inv[n]=qpow(fact[n],mod-2);
    for(int i=n-1;i>=1;i--)inv[i]=inv[i+1]*(i+1)%mod;
    for(int i=1,x;i<=n;i++){
        std::cin>>x;
        bl[i]=std::max(0,x-K);
        br[i]=std::min(x+K,i);
    }
    f[0][0][0]=1,sum[0][0][0]=1;
    for(int i=1,p=1;i<=n;i++,p=i&1){
        for(int j=0;j<=i;j++){
            for(int k=bl[i];k<=br[i]&&k<=j;k++){
                f[p][j][k]=(f[p][j][k]+f[p^1][j][k]*j%mod)%mod;
                if(j!=0)f[p][j][k]=(f[p][j][k]+f[p^1][j-1][k])%mod;
                if(j!=0&&k!=0)f[p][j][k]=(f[p][j][k]+sum[p^1][j-1][std::min(k-1,br[i-1])]*inv[j-k]%mod)%mod;
                sum[p][j][k]=f[p][j][k]*fact[j-k]%mod;
                if(k!=0)sum[p][j][k]=(sum[p][j][k]+sum[p][j][k-1])%mod;
            }
        }
        for(int j=0;j<=i-1;j++)
            for(int k=bl[i-1];k<=br[i-1]&&k<=j;k++)f[p^1][j][k]=sum[p^1][j][k]=0;
    }
    ll ans=0;
    for(int j=0;j<=n;j++)
        for(int k=bl[n];k<=br[n]&&k<=j;k++)
            ans=(ans+f[n&1][j][k]*fact[n-k]%mod*inv[n-j]%mod)%mod;
    std::cout<<ans<<"\n";
    return 0;
}
2023/8/8 12:35
加载中...