数组与字符串2 - 搜索插入位置

搜索插入位置

一 题目
给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
二 解法
思路：因为数组为升序排列数组，因此若target等于nums[index]，则index为目标值的索引。若target小于nums[index]，则index为即将插入的位置。

class Solution {
    public int searchInsert(int[] nums, int target) {
        int index = -1;
        for(int i=0;i<nums.length;i++){
            if(target <= nums[i]){
                index = i;
                break;
            }
        }
        if(index == -1 && target>nums[nums.length-1]){
            index = nums.length;
        }
        return index;
    }
}

三 优化
二的解法为暴力搜索，需要O(n)的时间复杂度，但是使用二分查找的话，可以降到O(logn)的时间复杂度。

class Solution {
    public int searchInsert(int[] nums, int target) {
        if(nums == null || nums.length == 0){
            return 0;
        }else if(target > nums[nums.length-1]){
            return nums.length;
        }else{
            int low = 0;
            int high = nums.length-1;
            int mid = 0;
            int flag = -1;
            while(low<=high){
                mid = (low + high)/2;
                if(target==nums[mid]){
                    flag = mid;
                    break;
                }else if(target<nums[mid]){
                    high=mid-1;
                }else{
                    low = mid+1;
                }
            }
            if(flag == -1 && target<nums[low]){
                flag = low;
            }
            if(flag == -1 && target>nums[low]){
                flag = low+1;
            }
            return flag;
        }
        
    }
}