当前位置: 代码迷 >> 综合 >> 找出升序排序数组第一个小于目标值的所有下标,第一个大于目标值的所有下标——四次二分搜索
  详细解决方案

找出升序排序数组第一个小于目标值的所有下标,第一个大于目标值的所有下标——四次二分搜索

热度:87   发布时间:2024-02-22 12:23:02.0

题目:找出升序排序数组第一个小于目标值的所有下标,第一个大于目标值的所有下标——四次二分搜索

思路:

1. 第一次二分:找目标值的最左下标

2. 第二次二分:找小于目标值的第一个数的值的最左边界

3. 第三次二分:找第一个大于目标值的下标

4. 第四次二分:找第一个大于(第一个大于目标值的值)的下标

例如: {-1,2,2,3,3,3,5,5,5,6,7,8,10};  //结果应该为 1,2,6,8

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
//找出第一个大于目标数的所有下标,找出第一个小于目标数的所有下标
int findLeft(vector<int>& nums,int target)
{int len=nums.size();int left=0;int right=len;while(left<right){int mid=(left+right)/2;if(nums[mid]>=target){right=mid;}else{left=mid+1;}}return left;
}
int findLeft(vector<int>& nums,int target,int left,int right)
{while(left<right){int mid=(left+right)/2;if(nums[mid]>=target){right=mid;}else{left=mid+1;}}return left;
}
int findRight(vector<int>& nums,int target)
{int len=nums.size();int left=0;int right=len;while(left<right){int mid=(left+right)/2;if(nums[mid]>target){right=mid;}else{left=mid+1;}}return left;
}
int findRight(vector<int>& nums,int target,int left,int right)
{while(left<right){int mid=(left+right)/2;if(nums[mid]>target){right=mid;}else{left=mid+1;}}return left;
}
vector<int> searchRange(vector<int>& nums, int target) {vector<int>result={-1,-1,-1,-1};    //返回第一个小于目标数的数开始下标,结束下标。返回第一个大于目标数的数开始下标,结束下标。int len=nums.size();int target_index_left=findLeft(nums,target);   //首先找到目标数的左边界cout<<target_index_left<<endl;if(target_index_left==len||nums[target_index_left]!=target)return result;if(target_index_left!=0){int little_left_index=findLeft(nums,nums[target_index_left-1],0,target_index_left);    //找到第一个小于目标数的数开始下标int little_right_index=target_index_left-1;result[0]=little_left_index;result[1]=little_right_index;}int index_right_first=findRight(nums,target); //找到第一个大于目标数的数下标if(index_right_first!=len){int bigger_left_index=index_right_first;int bigger_right_index=findRight(nums,nums[index_right_first],index_right_first,len)-1; //找到第一个大于(第一个大于目标数的数的值)的下标result[2]=bigger_left_index;result[3]=bigger_right_index;}return result;
}
int main()
{vector<int>nums={-1,2,2,3,3,3,5,5,5,6,7,8,10};  //结果应该为 1,2,6,8vector<int>result=searchRange(nums,3);//输出第一个小于目标数的数开始下标,结束下标。第一个大于目标数的数开始下标,结束下标。cout<<result[0]<<" "<<result[1]<<" "<<result[2]<<" "<<result[3]<<endl;return 0;
}

 

  相关解决方案