字节跳动笔试题02万万没想到之抓捕孔连顺

我叫王大锤，是一名特工。我刚刚接到任务：在字节跳动大街进行埋伏，抓捕恐怖分子孔连顺。和我一起行动的还有另外两名特工，我提议

1. 我们在字节跳动大街的N个建筑中选定3个埋伏地点。
2. 为了相互照应，我们决定相距最远的两名特工间的距离不超过D。

我特喵是个天才! 经过精密的计算，我们从X种可行的埋伏方案中选择了一种。这个方案万无一失，颤抖吧，孔连顺！
……
万万没想到，计划还是失败了，孔连顺化妆成小龙女，混在cosplay的队伍中逃出了字节跳动大街。只怪他的伪装太成功了，就是杨过本人来了也发现不了的！

请听题：给定N（可选作为埋伏点的建筑物数）、D（相距最远的两名特工间的距离的最大值）以及可选建筑的坐标，计算在这次行动中，大锤的小队有多少种埋伏选择。
注意：

1. 两个特工不能埋伏在同一地点
2. 三个特工是等价的：即同样的位置组合(A, B, C) 只算一种埋伏方法，不能因“特工之间互换位置”而重复使用

参考代码

#include
#include
using namespace std;
 
long long C_n_k(int n, int k) {
 if (n < k)return 0;
 vector cur(n + 1, 0);
 vector > c(n + 1, cur);
 c[1][0] = c[1][1] = 1;
 for (int i = 2; i <= n; i++) {
  for (int j = 0; j <= i; j++) {
 
   if (j == 0 || j == i)
    c[i][j] = 1;
   else
    c[i][j] = c[i - 1][j - 1] + c[i - 1][j];
  }
 }
 return c[n][k];
}
 
long long C(long long n) {
 return n * (n - 1) / 2;
}
 
int main() {
 long long  N, D;
 cin >> N >> D;
 long long cnt = 0;
 vector pos(N, 0);
 cin >> pos[0] >> pos[1];
 for (int i = 2, j=0; i < N; i++) {
  cin >> pos[i];

   while (pos[i] - pos[j] >D) {
    j++;
   }
   cnt += C(i - j);
 }
 cout << cnt%99997867 ;
 system("pause");
}

版本2

#include  
#include 
using namespace std;

long long C(long long n){
    return (n-1) * n / 2;
}

int main()
{
    long long n, d, count = 0;
    cin>> n>> d;
    vector v(n);
    for (int i = 0, j = 0; i < n; i++) {
        cin>> v[i];
        while (i >= 2 && (v[i] - v[j]) > d) {
            j++;
        }
        count += C(i - j);
    }
    cout << count % 99997867; 
    return 0;
}