春季赛补题 音符

题目描述

大G在拥有强大的算法知识储备的同时，还会创作悦耳动听的乐曲，因此深得系里学妹的追捧。

大G手上有n个音符，每个音符都有一个音高值。如果一个乐曲中有两个音符的音高值相差过大，那么这一首乐曲就会十分不协调，因此大G创作乐曲时，同一首乐曲中绝对不会出现两个音符的音高相差大于k。

大G这会儿要创作两首乐曲来取悦小Q，他希望在里面加入的音符越多越好。请你告诉他，在他即将创作的这两首乐曲中，总共最多能放入多少音符。

输入

第一行两个正整数n，k。

接下来n行每行一个正整数，第i行的正整数ai表示第i个音符的音高。

输出

输出一个正整数，表示大G能在两首乐曲中放入的最多的音符数量。

样例输入 Copy

【样例1】
10 3
12
17
6
3
18
12
18
14
4
13
【样例2】
5 5
11
13
15
6
19
【样例3】
6 4
8
4
2
19
5
6

样例输出 Copy

【样例1】
7
【样例2】
4
【样例3】
5

提示

样例 1 解释：第一首乐曲放了音高为$12,12,13,14$的音符，第二首乐曲放了音高为$17,18,18$的音符，总共放了$7$个音符，不存在更优解。
 

找两个区间，几个连续区间头尾都可能有重复

先找一次连续最大，去掉这些音符再找，可能会漏。可能吧？？？

暴力枚举所有的组合的时候，从尾巴开始找之前的组合里面的最大值。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,k,m[500009],what[500009],big[500009],len[500009];

int main(){
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;++i)scanf("%d",&m[i]);
	sort(m,m+n);
	for(int i=n-1;i>=0;--i){
		int l=0,r=i,mid,p=i;
		while(l<=r){
			mid=(l+r)>>1;
			if(m[i]-m[mid]>k){
				l=mid+1;
			}
			else{
				p=mid;
				r=mid-1;
			}
		}
		what[i]=p;
		len[i]=i-p+1;
	}
	big[0]=len[0];
	for(int i=1;i<n;++i){
		big[i]=len[i]>big[i-1]?len[i]:big[i-1];
	}
	int ans=0,k;
	for(int i=n-1;i>0;--i){
		k=len[i]+big[what[i]-1];
		ans=ans>k?ans:k;
	}
	printf("%d",ans);
	return 0;
}