字节真题：万万没想到之抓捕孔连顺

牛客链接：https://www.nowcoder.com/question/next?pid=16516564&qid=362290&tid=49703321

直接上代码，记录下我的做题经过

// 应该至少需要3个while循环
// 第一个循环：固定第一个特工位置，另外两个特工往右找
// 第二个循环：固定第二个特工位置，如果与第一个特工保持在最大范围内，向右招最后一个特工,否则，返回第一个循环第一个特工位置右移
// 第三个循环：如果与第一个特工的位置超过最大范围，返回第一个循环（不用和第二个特工位置做比较）
// 结果，通过了2/10的用例，出现超时问题

// 第二种优化方法：时间复杂度O(n)
// 第一个循环：固定第一个特工位置
// 第二个循环，不断往后找第三个特工位置，直到其位置超出最大距离，根据两个位置的索引，得到区间内元素的个数
// 假设元素个数为x，那么这个区间内可选的方案数为：C(x,3)
// 结果，通过了2/10的用例，未出现超时问题

// 第三种方法：时间复杂度O(n)
// 第二种方法方向是正确的，但没考虑仔细，比如：1 2 3 4 5 6 最大距离为4；当ridx指向6时，fidx指向4，少了2 3 6和 2 4 6的组合
// 看了下题解，应该固定第一个特工，其位置不断向右滑动，这样每个区间都是独立的，如第一个区间一定包括1，后面的区间一定不包括1
// 每个区间区C(x,2)
#include<iostream>
#include<vector>
using namespace std;

#define MAX_VAL 99997867

long long getRes(vector<long long>init, long long k) {
    long long res = 0;
    long long fidx = 0, ridx=2;
    long long len = init.size();
    while (fidx < len - 2) {
        // ridx = fidx + 2;   // 后面的指针不需要往回缩了
        while (ridx < len && (init[ridx] - init[fidx]) <= k) {
            ridx++;
        }
        // ridx指向范围区间的下一个,计算c(ridx-1-fidx,2)
        long long tmp = ridx -1- fidx;
        res = (res+(tmp - 1) * tmp / 2)%MAX_VAL;
        fidx++;          // 第一个特工右移
    }
    return res % MAX_VAL;
}

int main() {
    long long d, k;
    cin >> d >> k;
    vector<long long> init;
    for (long long i = 0; i < d; ++i) {
        long long tmp;
        cin >> tmp;
        init.push_back(tmp);
    }
    long long res = getRes(init, k);
    cout << res;
    return 0;
}