感觉自己对二分法还是没有很好掌握,主要在于边界问题,只会基本的搜索,如果要搜索比目标值大的第一个索引或者比目标值小的最后一个索引(或者换一些花里胡哨的题目)就完全不会了。
class Solution {
public:int searchInsert(vector<int>& nums, int target) {int left=0;int right=nums.size()-1;int result=0;while(left<=right){int mid=(left+right+1)/2;if(nums[mid]<target){left=mid+1;result=left;}else if(nums[mid]>target) right=mid-1;else return mid;}return result;}
};
解法如上,result先默认为0(如果二分一直往后撤,最终left甚至没动一次就说明第一个元素都比target大),接着二分搜索,注意只有当left右移时才要将result变为右移后的left,因为这时候mid对应的值一定比target小,如果mid是比target小的最后一个那么结果就是mid+1。
所以result只与left有关,甚至可以不设置result,最后返回left即可。