搜索插入位置

49 阅读1分钟

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。力扣题目

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

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

 

提示:

  • 1 <= nums.length <= 104
  • -104 <= nums[i] <= 104
  • nums 为 无重复元素 的 升序 排列数组
  • -104 <= target <= 104

解题思路:使用二分法查找,查得到直接返回;

public class SearchInsert {


    public static void main(String[] args) {
//        int nums[] = {1,3,5,6,8};int target = 5;

        int nums[] = {1,3,5,6}; int target = 2;
//        int nums[] = {1,3,5,6}; int target = 7;
        int i = searchInsert(nums, target);
        System.out.println("------------"+i);
    }

    public static int searchInsert(int[] nums, int target) {
        int left = 0;
        int right = nums.length -1;
        while (left <= right) {
            int middle = left +(right -left ) / 2;//防止int溢出
            if(nums[middle] == target) {
                return middle;
            }else if(nums[middle] < target) {
                left = middle + 1 ;
            }else if(nums[middle] > target) {
                right = middle -1 ;
            }
        }
        return right+1;
    }

}