ACWING/1824. 钻石收藏家

输出格式

输出贝茜可以在展示柜中展示的钻石最大数量。

数据范围

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