整数二分详解

模板一

在升序、无重复元素的数组中，查找值为k的元素的数组下标

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int n = 10;

// 在给定数组中查找k值所在位置，若不存在则返回-1
// l为二分下界，r为二分上界
int binarySearch(int a[], int l, int r, int k){
    int mid;
    // 此处的循环条件为 l <= r 的原因：
    // 若l > r，即不为闭区间时，作为元素不存在的依据，直接返回-1
    while (l <= r){
        // 此处也可使用 mid = l+(r-l)/2; 语句避免l+r溢出int
        mid = (l+r)/2;
        if (a[mid] == k){
            return mid;
        }else if (a[mid] > k){
            r = mid - 1;
        }else{
            l = mid + 1;
        }
    }
    return -1;
}
int main(){
    int k;
    int a[n] = {1, 3, 4, 6, 7, 8, 10, 11, 12, 15};
    scanf("%d", &k);
    // 此处传入闭区间[0, n-1]
    int pos = binarySearch(a, 0, n-1, k);
    printf("%d\n", pos);
    return 0;
}

模板二

在升序、有重复元素的数组中，查找第一个大于等于k值的元素的数组下标

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int n = 6;

// 在给定数组中查找第一个大于等于k值的元素的数组下标
// 若该元素不存在，则返回假设它存在，应该在的位置
// l为二分下界，r为二分上界
int lowerBoundL(int a[], int l, int r, int k){
    int mid;
    while (l < r){
        mid = (l+r)/2;
        if (a[mid] >= k){
            // 此处是r = mid，这是由判断条件a[mid]有可能等于k决定的
            r = mid;
        }else{
            l = mid + 1;
        }
    }
    // 此处返回r或l皆可，因为r == l
    return l;
}

int main(){
    int k;
    int a[n] = {1, 2, 2, 3, 3, 4};
    while(~scanf("%d", &k)){
    	// 此处传入闭区间[0, n]
    	// 因为可能数组中所有值都小于k，则返回该元素的应在位置n
        int pos = lowerBoundL(a, 0, n, k);
        printf("%d\n", pos);
    }
    return 0;
}

模板三

在升序、有重复元素的数组中，查找最后一个小于等于k值的元素的数组下标

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int n = 6;

// 在给定数组中查找最后一个小于等于k值的元素的数组下标
// 若该元素不存在，则返回下标0
// l为二分下界，r为二分上界
int lowerBoundR(int a[], int l, int r, int k){
    int mid;
    while (l < r){
        // 注意此处对(l+r)/2做上取整，防止仅剩两个数时发生死循环
        // 注意：只在l = mid时才做如此处理
        mid = (l+r+1)/2;
        if (a[mid] <= k){
            l = mid;
        }else{
            r = mid-1;
        }
    }
    return l;
}

int main(){
    int k;
    int a[n] = {1, 1, 1, 3, 3, 4};
    while(~scanf("%d", &k)){
        // 此处传入区间[0, n-1]
        int pos = lowerBoundR(a, 0, n-1, k);
        // 由于查找元素不存在时也返回0，故要对返回的数组下标做判断
        if (pos != 0||(pos == 0 && a[0] <= k)){
            printf("%d\n", pos);
        }else{
            printf("-1\n");
        }
    }
    return 0;
}

使用模板二和模板三可以在数组中找到重复元素的区间

模板四

在升序、有重复元素的数组中，查找第一个大于k值的元素的数组下标

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int n = 6;

// 在给定数组中查找第一个大于k值的元素的数组下标
// 若该元素不存在，则返回假设它存在，应该在的位置
// l为二分下界，r为二分上界
int upperBound(int a[], int l, int r, int k){
    int mid;
    while (l < r){
        mid = (l + r)/2;
        // 相较于模板二，只是在此处去掉了等号
        if (a[mid] > k){
            r = mid;
        }else{
            l = mid + 1;
        }
    }
    return l;
}

int main(){
    int k;
    int a[n] = {1, 1, 1, 3, 3, 4};
    while(~scanf("%d", &k)){
        // 此处传入区间[0, n]
        // 原因见模板二
        int pos = upperBound(a, 0, n, k);
        printf("%d\n", pos);
    }
    return 0;
}

待补充…