首页 > 其他 > 详细

Leetcode 35. Search Insert Position

时间:2016-01-15 06:24:04      阅读:230      评论:0      收藏:0      [点我收藏+]

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


解题思路:

binary search 经典思路

四点要素:
1. start + 1 < end
2. start + (end - start) / 2

3. A[mid] ==, <, >

4. A[start] A[end] ? target 

把题目逻辑想清楚,其实就是找出多少数比target小的,也可以想成找第一个大于等于目标数的位置。

注意中间那块是模板。


Java code:
1. find the last position < target 
public class Solution {
    public int searchInsert(int[] nums, int target) {
        if (nums == null || nums.length == 0) {
            return -1;
        }
        
        int start = 0;
        int end = nums.length - 1;
        int mid;
        
        if(target < nums[0]) {
            return 0;
        }
        
        //find the last number less than target
        while (start + 1 < end) {
            mid = start + (end - start) / 2;
            if (nums[mid] == target) {
                end = mid;  
            } else if (nums[mid] < target) {
                start = mid;
            } else {
                end = mid;
            }
        }
        if (nums[end] == target) {
            return end;
        }
        if(nums[end] < target) {
            return end + 1;
        }
        if(nums[start] == target) {
            return start;
        }
        return start + 1;
    }
}

2. find the first position >= target

public class Solution {
    public int searchInsert(int[] nums, int target) {
        if (nums == null || nums.length == 0) {
            return -1;
        }
        
        int start = 0;
        int end = nums.length - 1;
        int mid;
        
        //find the first position >= target
        while (start + 1 < end) {
            mid = start + (end - start) / 2;
            if (nums[mid] == target) {
                end = mid;  
            } else if (nums[mid] < target) {
                start = mid;
            } else {
                end = mid;
            }
        }
        if (nums[start] >= target) {
            return start;
        }else if (nums[end] >= target) {
            return end;
        }else {
            return end+1;
        }
    }
}

Reference:

1. http://www.jiuzhang.com/solutions/search-insert-position/

Leetcode 35. Search Insert Position

原文:http://www.cnblogs.com/anne-vista/p/5132190.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!