1. 程式人生 > >leetcode 35. 搜尋插入位置

leetcode 35. 搜尋插入位置

給定一個排序陣列和一個目標值,在陣列中找到目標值,並返回其索引。如果目標值不存在於陣列中,返回它將會被按順序插入的位置。

你可以假設陣列中無重複元素。

示例 1:

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

示例 2:

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

示例 3:

輸入: [1,3,5,6], 7
輸出: 4

示例 4:

輸入: [1,3,5,6], 0
輸出: 0

一道典型的二分查詢題:AC程式碼如下:

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int start=0,end=nums.size()-1,middle=0;
        if(target>nums[end]) return end+1;
        while(start<=end){
        
        middle=start+(end-start)/2;
        if(target<nums[middle]){
            end=middle-1;
        }
        if(target>nums[middle]){
            start=middle+1;
        }
        if(target==nums[middle]){
            return middle;
        }
    }
     if(nums[start]>=target) return start;
        else  return start+1;
        
        
    return 0;
    }
};