首先了解二分查找的前提(必须是有序的数组)
下面看一个例题:
给你一个下标从 0 开始的整数数组 nums 以及一个目标元素 target 。
目标下标 是一个满足 nums[i] == target 的下标 i 。
将 nums 按 非递减 顺序排序后,返回由 nums 中目标下标组成的列表。如果不存在目标下标,返回一个 空 列表。返回的列表必须按 递增 顺序排列。
示例 1:
输入:nums = [1,2,5,2,3], target = 2
输出:[1,2]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 2 的下标是 1 和 2 。
示例 2:
输入:nums = [1,2,5,2,3], target = 5
输出:[4]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 5 的下标是 4 。
示例 3:
输入:nums = [1,2,5,2,3], target = 5
输出:[4]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 5 的下标是 4 。
1、首先进行排序
2、进行二分查找(分别找出左右端点)
看代码:(注意我写的怎么找左右端点,以后都可以用这个模板做二分的题)
class Solution { public: vector<int> targetIndices(vector<int>& nums, int target) { vector<int>ans; sort(begin(nums),end(nums)); int l = 0,r = nums.size()-1; int begin = -1; //左边界 while(l <= r) { int mid = (l + r)/2; if(nums[mid] < target) l = mid + 1; else{ r = mid - 1; begin = mid; } } //右边界 l = 0,r = nums.size()-1; int end = -1; while(l <= r) { int mid = (l + r)/2; if(nums[mid] <= target){ end = mid; l = mid + 1; }else r = mid - 1; } if(begin == -1 || end == -1) return ans; for(int i=begin;i<=end;i++) { ans.push_back(i); } return ans; } };