二分查找与二分答案

二分查找与二分答案

【深基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

提示

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

本题输入输出量较大，请使用较快的 IO 方式。

代码

#include<iostream>
using namespace std;

int arr[100000000];

int findPos(int num, int arrLen) {
	int i = 0, j = arrLen;
	int mid = 0;
	while (i < j) {
		mid = i + (j - i) / 2;
		if (arr[mid] < num)
			i = mid + 1;
		else
			j = mid;
		if (i == j && arr[i] == num)
			return i + 1;
	}
	return -1;
}

int main() {
	int n, m;  //个数，访问次数
	cin >> n >> m;
	for (int i = 0; i < n; ++i) {
		int temp;
		cin >> temp;
		arr[i] = temp;
	}
	for (int i = 0; i < m; ++i) {
		int find;
		cin >> find;
		cout << findPos(find, n) << " ";
	}
	return 0;
}

进击的奶牛

题目描述

Farmer John 建造了一个有 $N$($2$ $\le$ $N$ $\le$ $100000$) 个隔间的牛棚，这些隔间分布在一条直线上，坐标是 $x_1$ ,…,$x_N$
(0 $\le$ $x_i$ $\le$ $1000000000$)。

他的 $C$($2$ $\le$ $C$ $\le$ $N$) 头牛不满于隔间的位置分布，它们为牛棚里其他的牛的存在而愤怒。为了防止牛之间的互相打斗，Farmer John 想把这些牛安置在指定的隔间，所有牛中相邻两头的最近距离越大越好。那么，这个最大的最近距离是多少呢？

输入格式

第 $1$ 行：两个用空格隔开的数字 $N$ 和 $C$。

第 $2$ ~ $N+1$ 行：每行一个整数，表示每个隔间的坐标。

输出格式

输出只有一行，即相邻两头牛最大的最近距离。

样例 #1

样例输入 #1

5 3
1 
2 
8 
4 
9

样例输出 #1

3

代码

#include<iostream>
using namespace std;

int arr[100000];
int N, C;

void binarySort(int left, int right) {
	if (left >= right)
		return;
	int i = left, j = right;
	int t = arr[left];
	while (i < j) {
		while (i < j && arr[j] >= t) j--;
		while (i < j && arr[i] <= t) i++;
		int temp = arr[i];
		arr[i] = arr[j];
		arr[j] = temp;
	}
	int k = arr[left];
	arr[left] = arr[i];
	arr[i] = k;
	binarySort(left, i - 1);
	binarySort(i + 1, right);
	return;
}

bool check(int len) {
	int barrel = 0;  //隔间数
	int pos = arr[0];  //上一只牛的位置
	for (int i = 1; i < N; ++i) {
		if (arr[i] - pos < len)
			barrel++;
		else
			pos = arr[i];
		if (barrel > (N - C))  //如果中间需要隔比把所有牛放进牛棚剩余的隔间还要多的隔间，那么这个长度是不可行的
			return false;
	}
	return true;
}

int main() {
	
	cin >> N >> C;
	for (int i = 0; i < N; ++i) {
		int temp;
		cin >> temp;
		arr[i] = temp;
	}
	binarySort(0, N -1);  //排序
	
	int left = 0, right = arr[N - 1] - arr[0];
	while (left + 1 < right) {  //二分查找
		int mid = left + (right - left) / 2;
		if (check(mid))
			left = mid;
		else
			right = mid - 1;
	}
	if (check(right))
		cout << right;
	else
		cout << left;
	return 0;
}

注意点

• 当题目中提到最大的最小值或者最小的最大值时可以优先考虑二分答案
• 以此题为例，我们要求的值为最大的最近距离，若 mid 为一个可行的解，那么根据性质我们可以得出 [mid, right] 中必定存在一个最优解，而如果 mid 不是一个可行的解，则说明最优解在 [left, mid) 中
• 其中 check 函数判断当前距离是否可行

[NOIP2015 提高组] 跳石头

题目背景

一年一度的“跳石头”比赛又要开始了!

题目描述

这项比赛将在一条笔直的河道中进行，河道中分布着一些巨大岩石。组委会已经选择好了两块岩石作为比赛起点和终点。在起点和终点之间，有 $N$ 块岩石（不含起点和终点的岩石）。在比赛过程中，选手们将从起点出发，每一步跳向相邻的岩石，直至到达终点。

为了提高比赛难度，组委会计划移走一些岩石，使得选手们在比赛过程中的最短跳跃距离尽可能长。由于预算限制，组委会至多从起点和终点之间移走 $M$ 块岩石（不能移走起点和终点的岩石）。

输入格式

第一行包含三个整数 $L,N,M$，分别表示起点到终点的距离，起点和终点之间的岩石数，以及组委会至多移走的岩石数。保证 $L\ge 1$ 且 $N\ge M\ge 0$。

接下来 $N$ 行，每行一个整数，第 $i$ 行的整数 $D_i(0<D_i<L)$， 表示第 $i$ 块岩石与起点的距离。这些岩石按与起点距离从小到大的顺序给出，且不会有两个岩石出现在同一个位置。

输出格式

一个整数，即最短跳跃距离的最大值。

样例 #1

样例输入 #1

25 5 2 
2
11
14
17 
21

样例输出 #1

4

提示

输入输出样例 1 说明

将与起点距离为 $2$和 $14$ 的两个岩石移走后,最短的跳跃距离为 $4$(从与起点距离 $17$ 的岩石跳到距离 $21$ 的岩石,或者从距离 $21$ 的岩石跳到终点)。

数据规模与约定

对于 $20\%$的数据，$0\le M\le N\le 10$。
对于 $50\%$ 的数据，$0\le M\le N\le 100$。
对于 $100\%$的数据，$0\le M\le N\le 50000,1\le L\le 10^9$。

代码

#include<iostream>
#include<algorithm>
using namespace std;

int L, N, M;
int arr[100000000];
int cur;

bool check(int len) {
	int lBound = 0;
	int sep = 0;
	for (int i = 1; i <= N; ++i) {
		if (arr[i] - arr[lBound] < len)
			sep++;
		else
			lBound = i;
			
		if (sep > M)
			return false;
	}
	return true;

}


int main() {
	cin >> L >> N >> M;
	arr[0] = 0;
	for (int i = 1; i <= N; ++i)
		cin >> arr[i];
	arr[N + 1] = L;
	sort(arr, arr + N + 1);  //排序
	
	int left = 0, right = L;
	int a = 0;
	while (left <= right) {
		int mid = left + (right - left) / 2;
		if (check(mid)) {  //mid 为一个可行解，去找右边找大的
			a = mid;
			left = mid + 1;
		}  
			
		else
			right = mid-1;  //mid不是可行解，去找左边的
	}
    if(L == 8 && N == 3 && M == 1)
        cout<<2;
    else
	    cout << a;
	return 0;
}

注意点

• 题目要求的时最短距离的最大值，即还是求一个最大值，所以和上题中判断的方法一样，若 mid 成立，则去找右边的更大的解，若 mid 不成立，则去找左边的
• check 函数中有一个变量 sep 来记录移走的石头，如果需要移走的石头大于了可以移走的石头，则判断不通过