c++二分查找

洛谷原题

对于这个题目我们首先想到的肯定是二分查找。

二分查找代码

int binary_search(int start, int end, int key) {
  int ret = -1;  // 未搜索到数据返回-1下标
  int mid;
  while (start <= end) {
    mid = start + ((end - start) >> 1);  // 直接平均可能会溢出，所以用这个算法
    if (arr[mid] < key)
      start = mid + 1;
    else if (arr[mid] > key)
      end = mid - 1;
    else {  // 最后检测相等是因为多数搜索情况不是大于就是小于
      ret = mid;
      break;
    }
  }
  return ret;  // 单一出口
}

上面的代码实现的是查找一个给定的key值，如果能查找到，返回它的位置，查找不到的话返回-1.

但是单凭借这个功能是无法AC我们上面的那道题目的，因为上面的题目要求的是要输出这个数字

在序列中第一次出现的编号，而假如我们测试样例中有连续几个相同的数字例如：

1 3 3 3 5 7 9 11 13 15 15

那么上面的代码就无法实现将第一次出现的编号返回。

所以推荐我们使用c++内置函数：upper_bound 和 lower_bound

需要添加的头文件<algorithm>

函数介绍

lower_bound(int* first,int* last,val);

lower_bound(a.begin(),a.end(),x);

这个函数不仅可以用到已经排好序的普通数组中，还可以用于STL中的有序容器，如vector、set、map等

upper_bound 返回第一个大于搜索数的位置，而 lower_bound是第一个大于等于的数的位置。显然这道题用的是 lower_bound。但是当有序序列中不存在与给定值相同的元素，那么upper_bound 和 lower_bound函数的效果是一样的，它们都返回大于给定值的第一个位置。

返回结果（拿lower_bound举例）

返回从first开始的第一个大于或等于val的元素的地址。如果所有元素都小于val，则返回last的地址。

int a[8]={4,10,11,30,69,70,96,100};

int pos=lower_bound(a,a+8,11)-a;

cout<<pos;//输出结果为2

因为lower_bound(a,a+8,11)-a得到的结果是它们之间相差的元素个数，我们也可以在这里认为得到的是目标元素从0开始的下标位置

浅浅回顾一下

迭代器也可以相减

接着上面的说，当不存在这样的元素时：

pos = lower_bound( a, a + 8, 111) - number；

 pos = 8;

此时  lower_bound( a, a + 8, 111)返回的是最后一个元素后面一个的位置。

对于上面的洛谷题，完整的题解

#include <iostream>
// 边长数组，这样我们就不用担心数据装不下了
#include <vector>
using namespace std;
vector<int > vec;
int main() {
	int n, m;
	int t;
	cin >> n >> m;
	int index;
    // 向数组尾部放入数据，因为这个特性，vector也可以当成stack用
	while(n--) cin >> t, vec.push_back(t);
	while(m--) {
		cin >> t;
        // 在序列里查找目标数据
		index = lower_bound(vec.begin(), vec.end(), t) - vec.begin();
        // 如果目标数据找到了，输出答案，注意我们的数组下标是从0开始的
		if (t == vec[index]) cout << index + 1 << ' ';
        // 没找到记得输出-1哦
		else cout << -1 << ' ';
	}
	return 0;
}

参考文章：lower_bound()函数用法_lower_bound函数用法-CSDN博客