【每日一题】35.搜索插入位置

题目:https://leetcode-cn.com/problems/search-insert-position/

解法1,暴力
O(n) O(1)
注意两个条件 a.当数组中没有元素的时候 b.最后一个元素也不比 target 大

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        size_t i = 0;
        while (i < nums.size()) {
            if (nums[i] >= target) {
                return i;
            }
            i++;
        }
        return i == nums.size() ? i : -1;
    }
};

解法2,二分查找
O(logn) O(1)
注意循环终止条件、mid 的取值、以及 l+r 可能会越界!

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

EOF

发布了98 篇原创文章 · 获赞 91 · 访问量 4万+

猜你喜欢

转载自blog.csdn.net/Hanoi_ahoj/article/details/105376954
今日推荐