输出格式
输出贝茜可以在展示柜中展示的钻石最大数量。
数据范围
1≤N≤1000
0≤K≤10000
钻石的尺寸范围 [1,10000]
输入样例:
5 3
1
6
4
3
1
输出样例:
4
排序+双指针 常规解法
- 将数据进行排序,形成一个有序单调增加的数组。
- 然后左指针不变,向右移动右指针探寻直到超出k范围,记录包含的砖石个数
- 然后移动左指针到下一个位置,砖石个数减一,继续移动右指针探寻砖石并加数量
- 如此重复,取每次双指针中包含砖石数最大的为答案。
此方案的复杂度为排序的O(nlogn)决定
前缀和方案
还可以用空间换时间的方案,时间O(n),具体解法如下注释,较简单。
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <vector>
using namespace std;
int s[10000+10];
int main()
{
int n, k;
cin>>n>>k;
int in;
memset(s, sizeof s, 0);
for(int i = 0; i < n; ++i){
cin>>in;
s[in]++;
}
// 求前缀和
// s[i] 表示小于等于大小i的砖石数量总和
for(int i = 1; i <= 10000; ++i) s[i] += s[i-1];
// 求所有k区间内的最大值,即 s[i] - s[i-k-1]代表了一个k范围区间内砖石数量
int ans = 0;
for(int i = k+1; i <= 10000; ++i) ans = max(ans, (s[i] - s[i-k-1]));
cout<<ans<<endl;
return 0;
}