P1102 A-B 数对

P1102 A-B 数对

题目描述

出题是一件痛苦的事情！

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

好吧，题目是这样的：给出一串数以及一个数字 CC，要求计算出所有 A - B = CA−B=C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个整数 N, CN,C。

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

输出格式

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

输入输出样例

输入 #1

4 1
1 1 2 3

输出 #1

3

说明/提示

对于 75%75% 的数据，1 \leq N \leq 20001≤N≤2000。

对于 100%100% 的数据，1 \leq N \leq 2 \times 10^51≤N≤2×105。

保证所有输入数据都在 3232 位带符号整数范围内。

方法一

使用自己实现的lower_bound和upper_bound函数

#include<iostream>
#include<algorithm>
using namespace std;
long long int N, C;
long long int p[10000000];
long long int Lower_bound(long long int A[], long long int left, long long int right, long long int x)//返回第一个等于x的位置
{
	long long int mid, ans = 0;
	while (left <= right)
	{
		mid = left + (right - left) / 2;
		if (A[mid] >= x)
			right = (ans = mid) - 1;
		else
			left = mid + 1;
	}
	if (A[ans] == x)
		return ans;
	else
		return -1;
}
long long int Upper_bound(long long int A[], long long int left, long long int right, long long int x)//返回第一个大于x的位置
{
	long long int mid, ans = 0;
	long long int Asize = right;
	while (left < right)
	{
		mid = left + (right - left) / 2;
		if (A[mid] > x)
			right = (ans = mid);
		else
			left = mid + 1;
	}
	if (A[ans] > x)
		return ans;
	else
		return Asize;
}
int main()
{
	cin >> N >> C;
	for (long long int i = 0; i < N; i++)
		cin >> p[i];
	sort(p, p + N);
	long long int sum = 0;
	for (long long int i = 0; i < N - 1; i++)
		if (Lower_bound(p, 0, N, p[i] + C) != -1)
			sum += Upper_bound(p, 0, N, p[i] + C) - Lower_bound(p, 0, N, p[i] + C);

	cout << sum << endl;
}

方法二

使用STL中提供的lower_bound和upper_bound函数，注意返回值是指针

#include<iostream>
#include<algorithm>
using namespace std;
long long int N, C;
long long int p[10000000];
int main()
{
	cin >> N >> C;
	for (long long int i = 0; i < N; i++)
		cin >> p[i];
	sort(p, p + N);
	long long int sum = 0;
	for (long long int i = 0; i < N - 1; i++)
		if (lower_bound(p, p + N, p[i] + C) - p != N)
			sum += upper_bound(p, p + N, p[i] + C) - lower_bound(p, p + N, p[i] + C);

	cout << sum << endl;
}