题目链接
思路
- 已知:数据已排序
- 双指针 i,j
- i指向刚接收数据,j指向距离i最远并且距离小于d的那个位置,此时形成区间[i,j]
- 所以 相当于选择了最远那个并在在[i,j)中选出两个,此时C(2,i-j)
- 具体看代码
代码
#include <iostream>
#include <vector>
using namespace std;
// 求从n个中选出两个的组合数
long long C(long long n){
return (n-1) * n / 2;
}
int main()
{
long long n, d, count = 0;
cin>> n>> d;
vector<long long> v(n);
for (int i = 0, j = 0; i < n; i++) {
cin>> v[i];//选中v[i]这个位置的人,
//寻找区间
while (i >= 2 && (v[i] - v[j]) > d) {
j++;
}
//将区间中选出两个的组合数加入结果中
count += C(i - j);
count%=99997867;
}
return 0;
}