寻找序列的插入位置 Search Insert Position

来源:互联网 发布:设计院改制 知乎 编辑:程序博客网 时间:2024/06/09 16:44

题目: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. 题目源自于Leetcode。

[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0

思路:考虑两种情况,数组中已有要插入的元素、数组中没有要插入的元素。考虑特殊情况,在第一个数之前插入、在最后一个数之后插入。

代码一:循序查找。时间复杂度O(N)。

class Solution {public:    int searchInsert(int A[], int n, int target) {        int i;        for(i=0;i<n;i++)        {            if(A[i] < target)                continue;            else                return i;        }        if(i == n)            return i;    }};

代码二:二分查找。有序数组嘛,干嘛不用二分查找。题目说了数组中没有重复元素,如果有的话,找的是目标值范围的左端low_bound。时间复杂度O(logN)。

class Solution {public:    int searchInsert(int A[], int n, int target) {        int mid, left, right;        left = 0;        right = n-1;        while(left <= right)        {            mid = (left + right)/2;            if(A[mid] == target) //如果数组中已有此值,就在原位置插入                return mid;            else if(A[mid] > target)                right = mid - 1;            else                left = mid + 1;        }        return left; //如果数组中没有此值,最终left会是应插入位置    }};

原创粉丝点击