链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
题目描述
dd作为集训队的队长,一直掌管着集训室的空调遥控器,她需要调整温度使队员们更好地进入训练状态,已知集训室一共有nnn名队员,每位队员都有一个温度诉求a[i](1≤i≤n),当室内温度为K时,当且仅当∣a[i]−K∣≤p时,这个队员能够正常进入训练状态,否则就会开始躁动,作为队长,dd需要调整好温度,她想知道,在最佳情况下,最多有多少队员同时进入训练状态
输入描述:
第一行两个数n,p(1≤n,p≤1000000),含义如题面描述 接下来一行n个数a[i](1≤a[i]≤1000000)表示每个队员的温度诉求
输出描述:
输出一个数字,表示最多有多少队员同时进入训练状态
示例1
输入
6 2 1 5 3 2 4 6
输出
5
说明
温度调成3或4,都可以满足5名队员同时进入训练状态
解题思路:
二分
题目下,我要保证 温度在一个左闭右闭的区间 [ a[i]-p,a[i]+p ] 为了简便,在不更改原数组a[i]的情况下,我们以 a[i]-p 为温度基准,所能容忍的最大温度就是 a[i] ,这样我就不需要更改原 a[i] 数组的值,我们要保证的温度区间就变成了 [ a[i]-2*p,a[i] ] ,这两个区间在不需要考虑具体温度值是多少的情况下,在逻辑上是等价的
这样的话,我把温度值从小到大排序,数组 a 就变成了每个人所容忍的最大温度,那么我用每个人所能容忍的最小温度 a[i]-2*p ,如果这个最小温度小于这个人所容忍的最大温度,就说明这个人可以正常训练。用lower_bound进行二分查找,找到第一个大于等于这个值的位置,从找到的那个位置,一直到a数组的右边界,组成的左闭右闭区间中元素的个数就是ans,我们只需要遍历一遍n,更新出ans的最大值就可以了
下面附上ac代码
注意,使用lower_bound查找第一个大于等于的位置的时候,最好让数组下标从零开始,可以少很多计算边界的麻烦
#include <bits/stdc++.h>
using namespace std;
int a[1000005];
int main()
{
int n,p;
scanf("%d%d",&n,&p);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
sort(a+1,a+1+n);
int ans=1;
for(int i=2;i<=n;i++)
{
int pos=lower_bound(a+1,a+i+1,a[i]-2*p)-a;
ans=max(ans,i-pos+1);
}
printf("%d\n",ans);
//system("pause");
return 0;
}