【LeetCode】- Search Insert Position（查找插入的位置）

本文链接：https://blog.youkuaiyun.com/zdp072/article/details/24819843

本文介绍了一种在已排序数组中查找目标值插入位置的方法，包括常规遍历和高效的二分查找两种算法实现，并提供了具体示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[ 问题: ]

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order. You may assume no duplicates in the array.

翻译：给你一个排好序的数组和一个目标值，请找出目标值可以插入数组的位置。

[ 分析: ]

Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0

注意：一定要考虑一些特殊情况，如数组为null等。

[ 解法: ]

①. 常规解法：从数组索引为0的位置开始找，时间复杂度为 O(n) ，accepted

public class Solution {
	public int searchInsert(int[] A, int target) {
		if (A != null) {
			for (int i = 0; i < A.length; i++) {
				if (target == A[i] || target < A[i]) {
					return i;
				} 
			}
			return A.length;
		}
		return -1;
	}

	public static void main(String[] args) {
		int[] arr = { 1, 3, 5, 6 };
		System.out.println(new Solution().searchInsert(arr, 5)); // 5 -> 2
		System.out.println(new Solution().searchInsert(arr, 2)); // 2 -> 1
		System.out.println(new Solution().searchInsert(arr, 7)); // 7 -> 4
		System.out.println(new Solution().searchInsert(arr, 0)); // 0 -> 0
	}
}

②. 二分查找：时间复杂度log2n

前提条件：一定是有序数组。

public class Solution {
	public int searchInsert(int[] A, int target) {
		int mid;
		int low = 0;
		int high = A.length - 1;
		while (low < high) {
			mid = (low + high) / 2;
			if (A[mid] < target) {
				low = mid + 1; 
			} else if (A[mid] > target) {
				high = mid - 1;
			} else {
				return mid;
			}
		}
		
		return target > A[low] ? low + 1 : low;
	}

	public static void main(String[] args) {
		int[] arr = { 1, 3, 5, 6 };
		System.out.println(new Solution().searchInsert(arr, 5)); // 5 -> 2
		System.out.println(new Solution().searchInsert(arr, 2)); // 2 -> 1
		System.out.println(new Solution().searchInsert(arr, 7)); // 7 -> 4
		System.out.println(new Solution().searchInsert(arr, 0)); // 0 -> 0
	}
}