搜索插入位置

2019-04-19  本文已影响0人  Jimhou

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。

示例 1:
输入: [1,3,5,6], 5
输出: 2
示例 2:

输入: [1,3,5,6], 2
输出: 1
示例 3:

输入: [1,3,5,6], 7
输出: 4

示例 4:
输入: [1,3,5,6], 0
输出: 0

   public int searchInsert(int[] nums, int target) {
        if (nums.length == 0) {
            return 0;
        }
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] >= target)
                return i;
        }
        return nums.length;
    }

2:二分搜索,时间复杂度O(log N)

  public int binarySearchInsert(int[] nums, int target) {
        if (nums.length == 0) return 0;
        int l = 0, r = nums.length - 1;
        while (l <= r) {
            int m = l + ((r - l) >> 1);
            if (target == nums[m] || ((m == 0 || target > nums[m - 1]) && target < nums[m])) return m;
            if (target > nums[m]) l = m + 1;
            else r = m - 1;
        }
        return nums.length;
    }
上一篇 下一篇

猜你喜欢

热点阅读