牛客练习赛69 D

火柴排队

https://ac.nowcoder.com/acm/contest/7329/D

分析

考虑如果一个数有选择,那么可能会影响它的也必须选择。所以这个就是个计数 了。可以写个类似背包的转移来算出所有的可能方案,最后答案再除以所有方案数就好了。时间复杂度为 。当然你可以滚动数组优化空间。

一些细节

  • 表示可以选 个人时的合法方案数,由于从 转移过来的,所以可以滚动数组优化。
  • 关于逆元 保证 为质数,所以 ,那么乘上 ,等于模意义下除以

代码

#include<bits/stdc++.h>
using namespace std;
int read(){int x;scanf("%d",&x);return x;}
int n,D;
const int N = 5100,P = 998244353;
int A[N],f[2][N],top,C[N][N];
int fpow(int x,int b) {
    int a = 1;
    for(;b;b>>=1,x=1LL*x*x%P) 
    if(b&1) a = 1LL * a * x % P;
    return a;
}
void inc(int &x,int y) {(x+=y)%=P;}
int main()
{
    n = read();D = read();
    for(int i = 1;i <= n;i++) A[i] = read();sort(A+1,A+1+n);
    f[0][0] = 1;
    for(int l = 1,r = 0;l <= n;l = r + 1){
        r = l;while(A[r] >= A[r+1] - D && r < n) r++;
        top++;
        for(int i = 0;i <= n;i++) {
            for(int j = 0;j <= r-l+1;j++) {
                inc(f[top&1][i+j],f[top-1&1][i]); 
            }
        } 
        memset(f[top-1&1],0,sizeof(f[top-1&1]));
    }
    C[1][1] = 1;
    for(int i = 2;i <= n+1;i++) {
        for(int j = 1;j <= i;j++) {
            C[i][j] = (C[i-1][j-1] + C[i-1][j]) % P;
        }
    }
    for(int i = 1;i <= n;i++) {
        printf("%d\n",(1LL * f[top&1][i] * fpow(C[n+1][i+1],P-2)%P));
    }
    return 0;
}
全部评论

相关推荐

耀孝女:就是你排序挂了
点赞 评论 收藏
分享
评论
6
4
分享
牛客网
牛客企业服务