LeetCode 35 Search Insert Position (C,C++,Java,Python)

標籤：c   c++   java   python   leetcode   

Problem:

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

Solution:二分尋找，當找不到時l=r+1,所以根據最後一次l和r的變動來判定應該插入的位置，如果最後一次是l=mid+1，說明應該插入到mid+1的位置，如果最後一次是r=mid-1，則說明應該插入到mid的位置，具體可舉例自己手畫一下。
題目大意：給一個有序數組和一個目標整數，要求找到目標整數在數組中出現的位置，如果沒有，則返回目標整數插入數組後的位置。
Java原始碼（340ms）：

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

C語言原始碼（7ms）：

int searchInsert(int* nums, int numsSize, int target) {    int l=0,r=numsSize-1,pos=0,mid;    while(l<=r){        mid=(l+r)>>1;        if(nums[mid]==target)return mid;        else if(target>nums[mid]){            l=mid+1;pos=mid+1;        }else{            r=mid-1;            pos=mid;        }     }    return pos;}

C++原始碼（8ms）：

class Solution {public:    int searchInsert(vector<int>& nums, int target) {        int l=0,r=nums.size()-1,pos=0,mid;        while(l<=r){            mid=(l+r)>>1;            if(nums[mid]==target)return mid;            else if(target<nums[mid]){                r=mid-1;pos=mid;            }else{                l=mid+1;pos=mid+1;            }        }        return pos;    }};

Python原始碼（52ms）：

class Solution:    # @param {integer[]} nums    # @param {integer} target    # @return {integer}    def searchInsert(self, nums, target):        l=0;r=len(nums)-1;pos=0        while l<=r:            mid=(l+r)>>1            if target==nums[mid]:return mid            elif target>nums[mid]:l=mid+1;pos=mid+1            else:r=mid-1;pos=mid        return pos

LeetCode 35 Search Insert Position (C,C++,Java,Python)