牛客练习赛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; }