牛客网-字节跳动真题-万万没想到之抓捕孔连顺

题目链接

地址

思路

• 已知：数据已排序
• 双指针 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;
}