算法伴学笔记 Day 03 | 二分查找入门-CSDN博客

本文链接：https://blog.csdn.net/cheer2333/article/details/135745799

🏛️ Stay hungry, Stay foolish. --乔布斯斯坦福大学毕业演讲

🙌题单（List-二分查找 01）

由浅入深循序渐进
系列博客——
🟨算法伴学笔记 Day 01 | DFS入门
🟪算法伴学笔记 Day 02 | DFS入门（2）

Label 简单二分 边界查找
Level 普及-

参考资料 oiwiki - 二分查找

A 题详解

🟢原题回顾🟢

📋A 洛谷 P2249 【深基13.例1】查找

【深基13.例1】查找

题目描述

输入 $n$ 个不超过 $10^9$ 的单调不减的（就是后面的数字不小于前面的数字）非负整数 $a_1,a_2,\dots,a_{n}$ ，然后进行 $m$ 次询问。对于每次询问，给出一个整数 $q$ ，要求输出这个数字在序列中第一次出现的编号，如果没有找到的话输出 $- 1$ 。

输入格式

第一行 $2$ 个整数 $n$ 和 $m$ ，表示数字个数和询问次数。

第二行 $n$ 个整数，表示这些待查询的数字。

第三行 $m$ 个整数，表示询问这些数字的编号，从 $1$ 开始编号。

输出格式

输出一行， $m$ 个整数，以空格隔开，表示答案。

样例 #1

样例输入 #1

11 3
1 3 3 3 5 7 9 11 13 15 15
1 3 6

样例输出 #1

1 2 -1

提示

数据保证， $\leq n \leq 10^6$ ， $\leq a_i,q \leq 10^9$ ， $\leq m \leq 10^5$

原题回顾 END

🟢详解🟢

🟦从一般思路开始 —— 朴素的遍历查找

时间复杂度 O(mn)

---主要部分的代码---
for(int i=1;i<=m;i++){
        bool flag = false;
        for(int j=1;j<=n;j++){
            if(arr[j]==q[i]){
                printf("%d ",j);
                flag=true;
                break;
            }
        }
        if(!flag) printf("-1 ");
    }

在这里插入图片描述
太慢了遍历显然是不行滴🤔

🟪手写一个二分查找

时间复杂度O(n logm) 时间复杂度运行差异
对于较大数据而言二分查找具有显著的优势

---主要部分代码---
void find(int x)
{
    
    int l=1,r=n,mid=(l+r)/2;

    while(l<r){
        mid=(l+r)/2;
        if(arr[mid]>=x)
            r=mid;
        else
            l=mid+1;
    }
    
    if(arr[l]==x) printf("%d ",l);
    else printf("-1 ");
}

在这里插入图片描述
注意此题与传统简单二分相比有一定差异因为需要求出的是第一个target的位置
可以有以下两种不同写法
🟦🟦l<r（推荐）

while(l<r){
        mid=(l+r)/2;
        if(arr[mid]>=x)
            r=mid;  
        else
            l=mid+1;

🟪🟪l<=r

while(l<=r){
        mid=(l+r)/2;
        if(arr[mid]>=x)
            r=mid-1;
        else
            l=mid+1;
    }

⭐边界问题处理参考博客二分查找与边界二分查找（寻找边界）

在这里插入图片描述

🟩（另解）使用C++ STL

STL相关知识可见于 oiwiki - STL 算法
此处使用函数 lower_bound 获取范围内第一个大于等于目标数的地址

---完整AC代码---
#include<iostream>
#include<algorithm>

using namespace std;

const int N = 1e6 + 10;
const int Q = 1e5 + 10;

int n,m;
int arr[N];
int q[Q];

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&arr[i]);
    }
    for(int i=1;i<=m;i++){
        scanf("%d",&q[i]);
    }

    for(int i=1;i<=m;i++){
        int t = lower_bound(arr+1,arr+1+n,q[i]) - arr; //函数返回的是地址 正确解要减去数组首地址
        if(arr[t]==q[i]) printf("%d ",t);
        else printf("-1 ");
    }
}