35. Search Insert Position

题目描述

输入：一个排序好的数组nums，一个int target
返回：如果target在nums中存在，则返回下标。如果不存在，则返回target应该插入的位置的下标。
示例1：
Input: [1,3,5,6], 5
Output: 2
示例2：
Input: [1,3,5,6], 2
Output: 1
示例3
Input: [1,3,5,6], 0
Output: 0

分析

在排序的数组中找数值，二分搜索法。套用模板。
如果能找到target，直接返回index。
如果target不存在，那么返回的index 一定是 nums[index]>target的最小的下标。所以g(x)=nums[x]-target>0

	public int searchInsert(int[] nums, int target) {
        if(nums==null || nums.length==0) return 0;
        int n = nums.length;
        int l = 0,r = n-1;
        
        while(l<=r){
            int mid = l + (r-l)/2;
            if(nums[mid] == target){
                return mid;
            }
            if(nums[mid]>target){
                r = mid-1;
            }else{
                l = mid+1;
            }
        }
        return l;
    }
}

相似题目

34 Find First and Last Position of Element in Sorted Array

输入：一个升序数整数组nums，整数target。
输出：target在数组中的左边界和右边界。如果不存在返回[-1,-1]。

在数组中找目标值target的左边界，右边界，返回。
找左边界，相当于查找>=target的最小值。也就是g(x)=nums[x]-target>=0。需要考虑的是当target比数组中所有元素小，那么l=0，这个时候需要判断l=0是因为条件退出，还是等于target。同时也要考虑target比数组中所有元素大，这时候l=nums.length。

找右边界，相当于查找>target的最小值，结果再减1。也就是g(x)=nums[x]-target>0。找到最小的m,nums[m]>target，结果应该是m-1。不可以直接返回m-1，nums[m]>target,但是nums[m-1]可能等于target，也可能小于target。再考虑target比所有元素小，那么l=0，将不能再减1。再考虑target比所有元素大，l=nums.length。

class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] r = new int[]{-1,-1};
        int leftIndex = findTargetLeft(nums,target);
        if(leftIndex!=-1){
            r[0] = leftIndex;
            r[1] = findTargetRight(nums,target);
        }
        return r;
    }
    
    private int findTargetLeft(int[] nums, int target){
        int l =0,r = nums.length-1;
        while(l<=r){
            int middle = l + ((r-l)>>1);
            if(nums[middle] >= target){
                r = middle-1;
            }else{
                l = middle + 1;
            }
        }
        return l>= nums.length?-1:(nums[l]==target?l:-1);
    }
     private int findTargetRight(int[] nums, int target){
        int l = 0,r = nums.length-1;
        while(l<=r){
            int middle = l + ((r-l)>>1);
            if(nums[middle] > target){
                r = middle-1;
            }else{
                l = middle + 1;
            }
        }
        if(l<=0) return -1;
        return nums[l-1]==target?l-1:-1;
    }
}