牛客網-字節跳動真題-萬萬沒想到之抓捕孔連順

題目鏈接

地址

思路

• 已知：數據已排序
• 雙指針 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;
}