小红在一条道路上闲逛,道路上有许多日落果,想要在日落之前捡到尽可能多的日落果。将道路视为一条数轴,小红从原点出发,道路上有n(1 <=n<= 50000)个日落果,每个日落果有一个坐标x[i](-100,000 ≤xi ≤100,000)且日落果的坐标各不相同,t(1≤t <100000)分钟之后将会日落。
输入
第一行:两个整数t,n。
第二行: n个整数,表示日落果的坐标x[i]。
输出
一个整数,小红能捡到的最多的日落果数量。
输入样例 1
25 14 16 8 -7 3 10 -15 -17 6 -12 14 -13 2 9 -5
输出样例 1
8
思路:想要捡最多,肯定不能一直折返,最多折返一次。
所以先将所有的排序。然后得到捡到日落果的数量在0~n
然后二分答案,mid=(l+r)/2
接下来检查mid个能不能捡到。能捡到的话,j记录答案,左边界=mid+1,在重新找。如果不能,则右边界=mid-1;
检查方法:
枚举右边界的位置。如果是从第一个开始捡,那么第x个的下标就是x,所以从x到n枚举右端点。
根据右端点,推出左端点。
如果左端点的值在原点的右边,说明一直往右找就行,那就看右端点的值是不是小于时间t,是的话就符合,不是重新找。
如果右端点的值小于原点,说明一直往左找。然后看左端点的绝对值是不是小于等于t,是的话符合。
最后一直没找到,那说明当前值x不行,捡不到,太多了。
代码:
#include<iostream>
#include<algorithm>
using namespace std;
int a[2000100];
int t,n;
bool check(int x){
for(int r=x;r<=n;r++){
int l=r-x+1;
if(a[l]>=0 ){
if(a[r]<=t)
return true;
}
if(a[r]<=0 ){
if(-a[l]<=t)
return true;
}
if(a[l]<0 && a[r]>0){
if(min(-a[l],a[r])+a[r]-a[l]<=t){
return true;
}
}
}
return false;
}
int main(){
cin>>t>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
}
sort(a+1,a+1+n);
int l=1,r=n,ans;
while(l<=r){
int mid=(l+r)/2;
if(check(mid)){
l=mid+1;
ans=mid;
}
else{
r=mid-1;
}
}
cout<<ans;
return 0;
}