某 SCOI 模拟赛 T3 c

题意

用 $n$ 个积木拼一个塔，第 $i$ 个积木长 $a_i$。给定 $d$，要求积木 $i$ 能放在 $j$ 上面当且仅当 $a_i-a_j\le d$。求合法的搭积木方案数。答案模 $998244353$。$n\le 7\times 10^5$

题解

先排序。

考虑搭好前 $i-1$ 小的积木后，第 $i$ 个积木能插入进什么地方：它上面是哪个积木不重要，只需要它下面的积木够宽或者它下面没有积木。于是拿个指针维护一下大小不小于 $a_i$ 的最小积木，枚举时乘一下即可。

#include<bits/stdc++.h>
using namespace std;
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	}
	return ans*f;
}
const int N=7e5+10,mod=998244353;
int a[N];
int main(){
	int n=getint(),d=getint();
	for(int i=1;i<=n;i++)a[i]=getint();
	sort(a+1,a+n+1);
	int l=1;
	int ans=1;
	for(int i=1;i<=n;i++){
		while(a[l]<a[i]-d)++l;
		ans=ans*1ll*(i-l+1)%mod;
	}
	cout<<ans;
	return 0;
}