ACM模板——二分查找(完整版)

简介: ACM模板——二分查找(完整版)

64种。理由如下:

对其进行分类:

取整方式:向下取整 向上取整 (共2种)

区间开闭:闭区间 左闭右开区间 左开右闭区间 开区间 (共4种)

问题类型:

对于不下降序列a,求最小的i,使得a[i] = key

对于不下降序列a,求最大的i,使得a[i] = key

对于不下降序列a,求最小的i,使得a[i] > key

对于不下降序列a,求最大的i,使得a[i] < key

对于不上升序列a,求最小的i,使得a[i] = key

对于不上升序列a,求最大的i,使得a[i] = key

对于不上升序列a,求最小的i,使得a[i] < key

对于不上升序列a,求最大的i,使得a[i] > key


(共8种)

综上所述,二分查找共有 2*4*8=64 种写法。

算法的目的是解决问题,下面以针对不下降序列a的4个问题为例,给出我认为效率较高,较为简洁的代码。

对于不下降序列a,n为序列a元素的个数,key为关键字:


1.求最小的i,使得a[i] = key,若不存在,则返回-1

int binary_search_1(int a[], int n, int key)
{
    int m, l = 0, r = n - 1; // 闭区间[0, n - 1]
    while (l < r)
    {
        m = l + ((r - l) >> 1); // 向下取整
        if (a[m] < key) l = m + 1;
        else r = m;
    }
    if (a[r] == key) return r;
    return -1;
}

2.求最大的i,使得a[i] = key,若不存在,则返回-1

int binary_search_2(int a[], int n, int key)
{
    int m, l = 0, r = n - 1; // 闭区间[0, n - 1]
    while (l < r)
    {
        m = l + ((r + 1 - l) >> 1); // 向上取整
        if (a[m] <= key) l = m;
        else r = m - 1;
    }
    if (a[l] == key) return l;
    return -1;
}

3.求最小的i,使得a[i] > key,若不存在,则返回-1

int binary_search_3(int a[], int n, int key)
{
    int m, l = 0, r = n - 1;//闭区间[0, n - 1]
    while (l < r)
    {
        m = l + ((r - l) >> 1);//向下取整
        if (a[m] <= key) l = m + 1;
        else r = m;
    }
    if (a[r] > key) return r;
    return -1;
}

4.求最大的i,使得a[i] < key,若不存在,则返回-1

int binary_search_4(int a[], int n, int key)
{
    int m, l = 0, r = n - 1;//闭区间[0, n - 1]
    while (l < r)
    {
        m = l + ((r + 1 - l) >> 1);//向上取整
        if (a[m] < key) l = m;
        else r = m - 1;
    }
    if (a[l] < key) return l;
    return -1;
}

实际上,对于3、4,也可以先判断解是否存在,再进行二分查找。

显然,上面的代码还不够简洁。下面给出我认为最简洁的代码:

// 1.
int ans = std::lower_bound(a, a + n, key) - a;
ans = (ans == n || a[ans] != key) ? -1 : ans;
// 2.
int ans = std::upper_bound(a, a + n, key) - a;
ans = (ans == 0 || a[ans - 1] != key) ? -1 : ans - 1;
// 3.
int ans = std::upper_bound(a, a + n, key) - a;
ans = (ans == n) ? -1 : ans;
// 4.
int ans = std::lower_bound(a, a + n, key) - a;
ans = (ans == 0) ? -1 : ans - 1;
目录
相关文章
|
11月前
|
编译器 C++ 容器
[C++随想录] 模版进阶(上)
[C++随想录] 模版进阶(上)
|
4月前
|
vr&ar
[leetcode ~模版] 周赛模版
[leetcode ~模版] 周赛模版
|
10月前
|
存储 测试技术 C++
C++模板元模板实战书籍讲解第一章题目讲解
C++模板元模板实战书籍讲解第一章题目讲解
63 0
C++模板元模板实战书籍讲解第一章题目讲解
|
10月前
|
机器学习/深度学习 设计模式 C++
C++模板元模板实战书籍讲解第一章(奇特的递归模板式以及小结和题目)
C++模板元模板实战书籍讲解第一章(奇特的递归模板式以及小结和题目)
69 0
|
11月前
|
C++
[C++随想录] 模版进阶(下)
[C++随想录] 模版进阶(下)
|
移动开发 C++
算法笔记(三)——二分查找(超详细,附带模板)
算法笔记(三)——二分查找(超详细,附带模板)
算法笔记(三)——二分查找(超详细,附带模板)
算法竞赛入门到进阶之八数码问题模板
算法竞赛入门到进阶之八数码问题模板
|
算法 Java Python
ACM 选手图解 LeetCode 二叉搜索树中的搜索
ACM 选手图解 LeetCode 二叉搜索树中的搜索
ACM 选手图解 LeetCode 二叉搜索树中的搜索