Codeforces Round #690 (Div. 3) E2. Close Tuples (hard version)

题目大意

对于给定序列$a_1,a_2,...,a_n$
求有多少对$i_1<i_2<...<i_m$满足条件：
$max(a_{i1},a_{i2},...,a_{im})-min(a_{i1},a_{i2},...,a_{im})\le k$

样例输入

4
3
1 4
2 3
3 6
4
1 2
2 3
3 5
4 5
5
1 2
3 8
4 5
6 7
9 10
5
1 5
2 4
3 5
3 8
4 8

样例输出

0
1
2
0

解题思路

显然是一个组合计数问题，由于$1\le a_i\le n$。
因此计数$a_i$ ，累加组合数$C_{sum[cnt[l],...,cnt[l+k]]}^m$。
要注意一下当前组合数中可能包含之前的贡献，减去这部分即可。
(一开始$memset$初始化整个$cnt$数组一直$tle$，发现$t$可以到$2e5$，只初始化$cnt[1...n]$即可，改下这就$AC$了)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ls rt<<1
#define rs rt<<1|1
const int N=2e5+6;
const int INF=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-9;
const double pi=acos(-1);
int cnt[N],pre[N];
ll C[N][105];
void prework(){
   C[0][0]=1;
   for(int i=1;i<=200005;++i){
       C[i][0]=1;
       for(int j=1;j<=100;++j) C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
   }
}
int main(){
   prework();
   int t;scanf("%d",&t);
   while(t--){
       int n,m,k;scanf("%d%d%d",&n,&m,&k);
       for(int i=1;i<=n;++i) cnt[i]=0;
       for(int i=1;i<=n;++i){
           int a;scanf("%d",&a);
           cnt[a]++;
       }
       for(int i=1;i<=n;++i) pre[i]=pre[i-1]+cnt[i];
       ll ans=0;
       int pr=0;
       for(int l=1;l<=n&&pr<n;++l){
           int r=min(l+k,n);
           if(cnt[l]==0) continue;
           int t1=pre[r]-pre[l-1];
           int t2=0;
           if(pr>=l) t2=pre[pr]-pre[l-1];
           ans=(ans+C[t1][m])%mod;
           if(pr>=l) ans=(ans-C[t2][m]+mod)%mod;	//减去重复累加的贡献
           pr=r;
       }
       printf("%lld\n",ans);
   }
   return 0;
}