组合计数 - Close Tuples (hard version) - CF 690 E2

组合计数 - Close Tuples (hard version) - CF 690 E2

题意：

$给定一个长度为n的序列a，$

$要从中挑选一个m元组(a_{i_1},a_{i_2},a_{i_3},...,a_{i_m})，i_1<i_2<...<i_m，满足$

$$max(a_{i_1},a_{i_2},a_{i_3},...,a_{i_m})-min(a_{i_1},a_{i_2},a_{i_3},...,a_{i_m})\le k$$

$计算满足这样的m元组，共有多少组，答案对10^9+7取模。$

输入：

$首行一个正整数T，表示测试数据数量，$

$接着一行三个正整数，依次为n,m,k$

$最后一行为n个正整数a_1,a_2,...,a_n$

输出：

$一个正整数，表示答案。$

数据范围：

$1\le T\le 2\cdot 10^5，1\le n\le 2\cdot 10^5，1\le m\le 100，1\le k\le n，1\le a_i\le n$

$所有测试样例的n的总和不超过2\cdot 10^5$

Example
input

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

output

2
6
1
20

---

分析：

计数思想和——《Prime Game - ICPC 2018 南京》相似

要理解：$条件:i_1<i_2<i_3<...<i_m的含义是，m元组不重复。$

$那么对于每一个元素a_i，目标应当是统计跟它差不超过k的所有数的个数，然后从中任选m-1个，$

$为了避免重复统计，我们对每个a_i，统计满足a_j\le a_i，且a_i-a_j\le k，的a_j的个数，$

$$\begin{gathered} 因此，我们先将序列排序，然后对每个a_i，二分查找第一个大于等于a_i-k的元素的位置po{s}_i， \\ 它对答案的贡献为C_{i-po{s}_i}^{m-1} \end{gathered}$$

$则答案为:$

$$\sum _{i=1}^n{C}_{i-po{s}_i}^{m-1}$$

时间复杂度： $O(n\cdot logn)$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>

using namespace std;

const int mod = 1e9+7;
const int N = 200010;

#define ll long long

int n, T;
int a[N];
int m, k;

ll quick_pow(int a,int b)
{
    ll res = 1;
    while(b)
    {
        if(b&1) res=res*a%mod;
        a=(ll)a*a%mod;
        b>>=1;
    }
    return res;
}

ll C(int n, int m)
{
    if(n<m) return 0;
    ll up = 1, down = 1;
    for(int i=n;i>n-m;i--) up=up*i%mod;
    for(int i=1;i<=m;i++) down=down*i%mod;
    down = quick_pow(down,mod-2);
    return up*down%mod;
}

int main()
{
    ios::sync_with_stdio(false);
    
    cin>>T;
    while(T--)
    {
        cin>>n>>m>>k;
        for(int i=1;i<=n;i++) cin>>a[i];
        sort(a+1,a+n+1);
        ll ans = 0;
        for(int i=1;i<=n;i++)
        {
            int pos = lower_bound(a+1,a+i+1,a[i]-k) - a;
            ans = (ans + C(i - pos, m - 1))%mod;
        }
        cout<<ans<<endl;
    }
    return 0;
}