A-B 数对 —二分查找—大于等于—大于

 

题目背景

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！

题目描述

给出一串正整数数列以及一个正整数 C，要求计算出所有满足 A−B=C的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个正整数 N,C

第二行，N 个正整数，作为要求处理的那串数。

输出格式

一行，表示该串正整数中包含的满足 A−B=C的数对的个数。

输入输出样例

输入 #1

4 1
1 1 2 3

输出 #1

3

 

 lower_bound和upper_bound的手写方法！！！

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
constexpr int N = 2e5+10;
int arr[N];
int n, m;
//找大于的
int serch1(int st,int x) {//左侧查找，若不符合则需要更新low；
	int low = 0, high = n - 1;
	while (low < high) {
		int mid = low + high >> 1;
		if (arr[mid] > x) high = mid;
		else low = mid + 1;
	}
	return low;
}
//找大于等于的
int serch2(int st, int x) {//左侧查找，若不符合则需要更新low；
	int low = 0, high = n - 1;
	while (low < high) {
		int mid = low + high >> 1;
		if (arr[mid] >= x) high = mid;
		else low = mid + 1;
	}
	return low;
}
signed main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	cin >> n >> m;
	for (int i = 0;i < n;i++) {
		cin >> arr[i];
	}
	sort(arr, arr + n);
	int sum = 0;
	for (int i = 0;i < n;i++) {
		int tar = arr[i]-m;//需要查找的值；
		sum += serch1(arr[i], tar) - serch2(arr[i], tar);
	}
	cout << sum << endl;
	return 0;
}