LeetCode 704. 二分查找

题目描述

704.二分查找
给定一个 n 个元素有序的（升序）整型数组 nums 和一个目标值 target ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。

示例 1:

输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4

示例 2:

输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1

代码

C

int search(int* nums, int numsSize, int target)
{
    int left = 0;
    int right = numsSize - 1;
    int mid = 0;
    int index = -1;

    while (left <= right) {
        // 取left和right中间位置，防止(left + right) / 2 越界
        // 这里要注意 >> 优先级小于 +
        mid = left + ((right - left) >> 1);
        if (nums[mid] > target) {
            right = mid - 1; // target 在左区间[left, mid-1]
        }
        else if (nums[mid] < target) {
            left = mid + 1; // target 在右区间[mid+1, right]
        } else {
            index = mid;
            break;
        }
    }

    return index;
}

go

func search(nums []int, target int) int {

    l := 0
    r := len(nums) - 1
    for l <= r {
        m := l + ((r - l) >> 1)
        if nums[m] < target {
            l = m + 1
        } else if nums[m] > target{
            r = m - 1
        } else {
            return m 
        }
    }

    return -1
}