P1102 A-B 数对

1.题目

题目链接

2.分析

首先先排序（为了后续二分查找方便）
依次对每个元素A，运用二分找到>=A+C的最小边界和>A+C的最小边界，两者之差便是所求的实数对个数

3.代码

1.ans开int (❌)

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

int Search(int a[], int n, int flag)  //找到》= 和  >的最小值
{
	int ans = 0;
	for (int i = 0; i < n; i++)
	{
		int index = a[i] + flag;
		int t0 = 0, t1 = 0;
		int l = 0, r = n - 1;
		while (l <= r)
		{
			int mid = l + ((r - l) >> 1);
			if (a[mid] >= index)
			{
				t0 = mid;
				r = mid - 1;
			}
			else
			{
				l = mid + 1;
			}
		}
		if (!t0)   //有可能找不到符合条件的
			t0 = n;
		l = 0, r = n - 1;  //重新赋值  
		while(l <= r)           //找>index的最小下标
		{
			int mid = l + ((r - l) >> 1);
			if (a[mid] > index)
			{
				t1 = mid;
				r = mid - 1;
			}
			else
			{
				l = mid + 1;
			}
		}
		if (!t1)
			t1= n;
		ans += t1 - t0;
	}
	return ans;
}
int main()
{
	int n, flag;
	cin >> n >> flag;
	int* a = new int[n];
	for (int i = 0; i < n; i++)
		cin >> a[i];
	sort(a, a + n);
	cout << Search(a,n, flag);
	delete[]a;
	return 0;
}

2.ans开long long (√)

#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
ll Search(int a[], int n, int flag)  //找到》= 和  >的最小值
{
	ll ans = 0;
	for (int i = 0; i < n; i++)
	{
		int index = a[i] + flag;
		int t0 = 0, t1 = 0;
		int l = 0, r = n - 1;
		while (l <= r)
		{
			int mid = l + ((r - l) >> 1);
			if (a[mid] >= index)
			{
				t0 = mid;
				r = mid - 1;
			}
			else
			{
				l = mid + 1;
			}
		}
		if (!t0)   //有可能找不到符合条件的
			t0 = n;
		l = 0, r = n - 1;  //重新赋值  
		while(l <= r)           //找>index的最小下标
		{
			int mid = l + ((r - l) >> 1);
			if (a[mid] > index)
			{
				t1 = mid;
				r = mid - 1;
			}
			else
			{
				l = mid + 1;
			}
		}
		if (!t1)
			t1= n;
		ans += t1 - t0;
	}
	return ans;
}

int main()
{
	int n, flag;
	cin >> n >> flag;
	int* a = new int[n];
	for (int i = 0; i < n; i++)
		cin >> a[i];
	sort(a, a + n);
	cout << Search(a,n, flag);
	delete[]a;
	return 0;
}

3.优化，利用upper_bound(),lower_bound()

#include <iostream>
#include <algorithm>   //upper_bound  lower_bound的头文件
using namespace std;
typedef long long ll;
int main()
{
	ll ans = 0;
	int n, flag;
	cin >> n >> flag;
	int* a = new int[n];
	for (int i = 0; i < n; i++)
		cin >> a[i];
	sort(a, a + n);   //必须排序，为了后面的二分查找
	for (int i = 0; i < n; i++)
		ans += ((upper_bound(a, a + n, a[i] + flag) - a) - (lower_bound(a, a + n, a[i] + flag) - a));
	cout << ans;
	delete[]a;
	return 0;
}

4.总结

通过此题加深了我对二分的理解，增强对数据的感知能力(开 long long)
比如：
while（r<=l） // =不要省掉!!
找到>= <某个数的下标 的方法
学到了upper_bound lower_bound函数
其他方法，后续学到了会重新补上

5.更新日志

2022.5.12 整理上传

欢迎交流、讨论、指正~
不正确、不理解之处欢迎评论留言~