LeetCode(035) Search Insert Position (Java)

最新推荐文章于 2025-04-21 22:09:55 发布

原创最新推荐文章于 2025-04-21 22:09:55 发布 · 661 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Arrays

InterviewQuestions 同时被 3 个专栏收录

183 篇文章

订阅专栏

DataStructure_Algorithm

175 篇文章

订阅专栏

Java

57 篇文章

订阅专栏

本文介绍了一个算法问题，即在一个已排序的数组中查找目标值的位置，若不存在则返回插入位置。通过具体示例展示了如何实现该算法，并分析了常见的错误及解决方法。

题目如下：

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

分析如下：

看上去简单，但是还是交了好几次才搞定。错误分析见下文代码:

我的代码:

//229ms
/*
Input:	[1,3], 4
Output:	1
Expected:2
Input:	[1,3], 2
Output:	2
Expected:1
Input:	[1,3], 3
Output:	2
Expected:1
*/
public class Solution {
    public int searchInsert(int[] A, int target) {
        if (A.length == 0) return 0;

        // NOTE1: 
        // 末尾3行已经包含A.lenth == 1时候的情况分析，所以这两行可以删掉。
        // if (A.length == 1 && A[0] >= target) return 0;
        // if (A.length == 1 && A[0] < target) return 1;
        int start = 0, end = A.length - 1, mid = start + (end - start)/2;

        // NOTE2: 
        // 一开始写成了 while(start < end) { 
        // 但是其实如果处理成start + 1 < end的话，就保证了最后一定有3个元素，所以A[start] ,A[mid], A[end]是3个各不相同的元素，
        // 这样就不用考虑最后只剩2个元素的情况下，A[mid] 其实和 A[start]相同的这个情况了。

        while(start  + 1 < end) {
            mid = start + (end - start) / 2;
            if (A[mid] == target) {
                return mid;
            } else if(A[mid] < target) {
                start = mid + 1;
            } else {
                end = mid - 1;
            }
        }
        
        // //BUG1  target <= A[start] 写成了 target <= start;
        // if (target <= start) return start;
        // else if (target > start && target <= end) return end;
        // else return end + 1;
        
        // //BUG2 边界条件 target <= A[end] 写成了 target < A[end]
        // if (target <= A[start]) return start;
        // else if (target > A[start] && target < A[end]) return end;
        // else return end + 1;
        
        if (target <= A[start]) return start;
        else if (target > A[start] && target <= A[end]) return end;
        else return end + 1;
    }
}