各类二分查找C++实现

1. 二分查找

// 二分查找，找到该值在数组中的下标，否则为-1
static int binarySerach(vector<int> array, int key) {
    int left = 0;
    int right = array.size() - 1;
    // 这里必须是 <=
    while (left <= right) {
        int mid = (left + right) / 2;
        if (array[mid] == key) {
            return mid;
        }
        else if (array[mid] < key) {
            left = mid + 1;
        }
        else {
            right = mid - 1;
        }
    }
    return -1;
}

2. 二分查找的变种

2.1 查找第一个与key相等的元素

// 查找第一个与key相等的元素
static int findFirstEqual(vector<int> array, int key) {
    int left = 0;
    int right = array.size() - 1;
    // 这里必须是 <=
    while (left <= right) {
        int mid = (left + right) / 2;
        if (array[mid] >= key) {
            right = mid - 1;
        }
        else {
            left = mid + 1;
        }
    }
    if (left < array.size() && array[left] == key) {
        return left;
    }
    return -1;
}

2.2 查找最后一个与key相等的元素

// 查找最后一个与key相等的元素
static int findLastEqual(vector<int> array, int key) {
    int left = 0;
    int right = array.size() - 1;
    // 这里必须是 <=
    while (left <= right) {
        int mid = (left + right) / 2;
        if (array[mid] <= key) {
            left = mid + 1;
        }
        else {
            right = mid - 1;
        }
    }
    if (right >= 0 && array[right] == key) {
        return right;
    }
    return -1;
}

2.3 查找第一个大于key的元素

// 查找第一个大于key的元素
static int findFirstLarger(vector<int> array, int key) {
    int left = 0;
    int right = array.size() - 1;
    // 这里必须是 <=
    while (left <= right) {
        int mid = (left + right) / 2;
        // 注：若查找第一个大于或等于key的元素，则if判断条件为>=
        if (array[mid] > key) {
            right = mid - 1;
        }
        else {
            left = mid + 1;
        }
    }
    if (left < array.size()) {
        return left;
    }
    return -1;
}

2.4 查找最后一个小于key的元素

// 查找最后一个小于key的元素
static int findLastSmaller(vector<int> array, int key) {
    int left = 0;
    int right = array.size() - 1;
    // 这里必须是 <=
    while (left <= right) {
        int mid = (left + right) / 2;
        // 注：若查找最后一个小于或等于key的元素，则if判断条件为<=
        if (array[mid] < key) {
            left = mid + 1;
        }
        else {
            right = mid - 1;
        }
    }
    if (right >= 0) {
        return right;
    }
    return -1;
}

2.5 总结

二分查找的变种需要考虑两个问题：

1. if 语句的判断符号
2. 返回 left 还是 right