P1102 A-B 数对

P1102 A-B 数对

题目描述

出题是一件痛苦的事情!

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

好吧,题目是这样的:给出一串数以及一个数字 CC,要求计算出所有 A - B = CAB=C 的数对的个数(不同位置的数字一样的数对算不同的数对)。

输入格式

输入共两行。

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

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

输出格式

一行,表示该串数中包含的满足 A - B = CAB=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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值