分析
考虑如果一个数有选择,那么可能会影响它的也必须选择。所以这个就是个计数 了。可以写个类似背包的转移来算出所有的可能方案,最后答案再除以所有方案数就好了。时间复杂度为
。当然你可以滚动数组优化空间。
一些细节
表示可以选
个人时的合法方案数,由于从
转移过来的,所以可以滚动数组优化。
- 关于逆元
保证
为质数,所以
,那么乘上
,等于模意义下除以
。
代码
#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;
}
京公网安备 11010502036488号