leetcode2.两数相加

题目

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4

提示:

• 1 <= nums.length <= 104
• -104 <= nums[i] <= 104
• nums 为 无重复元素 的 升序 排列数组
• -104 <= target <= 104

解答

1. 折半查找

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n = nums.size(); // 获取数组的大小
        int left = 0; // 初始化左指针
        int right = n - 1; // 初始化右指针
        
        while (left <= right) { // 当左指针不大于右指针时循环
            int mid = left + (right - left) / 2; // 计算中间位置，防止溢出

            if (nums[mid] == target) { // 如果中间的元素等于目标值
                return mid; // 返回中间位置
            } else if (nums[mid] < target) { // 如果中间元素小于目标值
                left = mid + 1; // 将左指针移动到中间位置的右侧，加1以避免循环
            } else { // 如果中间元素大于目标值
                right = mid - 1; // 将右指针移动到中间位置的左侧，减1以避免循环
            }
        }

        return left; // 如果没有找到目标值，返回应该插入的位置
    }
};