codeforces1462E2. Close Tuples (hard version)

https://codeforces.com/contest/1462/problem/E2#

注意到m元组中坐标是各不相同的，且只要求最大值和最小值，所以我们可以把a排序，反正m元组是无视坐标顺序的

然后我们就可以钦定这个m元组必选当前a[i]，现在要求>=a[i]-k中选出剩下m-1个，预处理一下搞个组合数就行了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxl=3e5+10;

int n,m,k,cnt,tot,cas;ll ans;
int a[maxl];
ll fac[maxl],inv[maxl];
bool vis[maxl];
char s[maxl];

inline ll c(int n,int r)
{
	if(r>n || r<0) return 0;
	assert(r==2);
	return 1ll*n*(n-1)/2;
}

inline void prework()
{
	//scanf("%d%d%d",&n,&m,&k);
	scanf("%d",&n);m=3;k=2;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	sort(a+1,a+1+n);
}

inline void mainwork()
{
	ans=0;
	for(int i=m;i<=n;i++)
	{
		int l=lower_bound(a+1,a+1+n,a[i]-k)-a;
		(ans+=c(i-l,m-1));
	}
}

inline void print()
{
	printf("%lld\n",ans);
}

int main()
{
	int t=1;
	scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}