leetCode -- 搜索插入位置

leetCode – 搜索插入位置

题目描述

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

你可以假设数组中无重复元素。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-insert-position

测试用例

示例 1:

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

示例 2:

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

题目分析

1. 可以用暴力循环法找出数组中元素的情况
2. 也可以用二分法求解，缩短循环次数

解题方法

一.暴力循环法

解题思路

1. 首先循环取出元素，当元素大于或等于指定值
2. 返回指定值，否则返回最后的长度

解题代码

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

测试结果

二. 二分法

解题思路

1. 先找到中间的值进行比较判断
2. 如果相等则直接返回
3. 如果指定值比中值大，往后比较
4. 如果比中值小，往左边比较

解题代码

class Solution {
    public int searchInsert(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        while(left <= right){
            int mid = (left + right) / 2;
            if(nums[mid] >= target){
                right = mid - 1;
            }else{
                left = mid + 1;
            }
        }
        return left;
    }
}

测试结果