比赛 期末考试3 评测结果 AAAAAAAAAAAAAAAAAAAA
题目名称 hope I can sort 最终得分 100
用户昵称 djyqjy 运行时间 2.483 s
代码语言 C++ 内存使用 72.68 MiB
提交时间 2026-02-11 11:58:25
显示代码纯文本
#include<bits/stdc++.h>
#define int long long
#define ll long long
#define pir pair<int,int>
#define fi first
#define se second
#define pb push_back
#define mp make_pair
using namespace std;
inline int re()
{
    int f=1,num=0;
    char c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') num=num*10+c-'0',c=getchar();
    return num*f;
}
int T;
void clear();
const int N=5010,MOD=998244353;
int n,m;
int dp[N][N];
int a[N];
int fac[N],ny[N];
int qpow(int a,int b=MOD-2)
{
    int mul=1;
    while(b)
    {
        if(b&1) mul=mul*a%MOD;
        b>>=1;a=a*a%MOD;
    }
    return mul;
}
int C(int a,int b)
{
    if(a<b) return 0;
    return fac[a]*ny[b]%MOD*ny[a-b]%MOD;
}
void work()
{
    freopen("hopeicansort.in","r",stdin);
    freopen("hopeicansort.out","w",stdout);
    clear();
    fac[0]=1;for(int i=1;i<=5005;i++) fac[i]=fac[i-1]*i%MOD;
    ny[5005]=qpow(fac[5005]);
    ny[0]=1;for(int i=5004;i>=1;i--) ny[i]=ny[i+1]*(i+1)%MOD;
    // cout<<C(2,2)<<' '<<C(3,0)<<endl;
    n=re();m=re();
    int num=0,cnt=0;
    for(int i=1;i<=n;i++)
    {
        a[i]=re();
        if(a[i]==1) num++;
    }
    for(int i=n;n-i+1<=num;i--) if(a[i]==1) cnt++;
    // cout<<num<<' '<<cnt<<endl;
    dp[0][cnt]=1;
    int tmp=qpow(n*(n-1)/2);
    for(int i=1;i<=m;i++)
    {
        for(int j=0;j<=num;j++)
        {
            dp[i][j]=(dp[i-1][j]*(MOD+1-(num-j)*(num-j)%MOD*tmp%MOD)%MOD+
                     (j>=1?dp[i-1][j-1]*(num-j+1)%MOD*(num-j+1)%MOD*tmp%MOD:0))%MOD;
        }
    }
    // cout<<dp[0][2]<<' '<<dp[1][2]<<' '<<dp[2][2]<<endl;
    printf("%lld\n",dp[m][num]%MOD);
}
signed main()
{
    T=1;
    while(T--) work();
    return 0;
}
void clear()
{
    return;
}