剑指offer刷题——字节跳动

字节的题就是个憨憨!!!纯粹是在考阅读理解,语法和知识点都相对基础,但是题目真的不知所云,难度总体不如鹅厂。

我叫王大锤,是一名特工。我刚刚接到任务:在字节跳动大街进行埋伏,抓捕恐怖分子孔连顺。和我一起行动的还有另外两名特工,我提议
1.我们在字节跳动大街的N个建筑中选定3个埋伏地点。
2.为了相互照应,我们决定相距最远的两名特工间的距离不超过D。

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

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

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

这个题考点是数据存放和滑动窗口算法,用long long来放数据才可以保证所有测试样例都通过。选取一个最远的特工,然后往前找两个位置(随机找),这不就是一个很简单的排列组合嘛,奥里给!

#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<long long> 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;
}

你可能感兴趣的:(c++,面向对象,STL)