A-B 数对

ada7_

已于 2024-02-29 23:10:48 修改

阅读量280

点赞数 3

文章标签：算法数据结构

于 2024-02-29 23:10:23 首次发布

本文链接：https://blog.csdn.net/ada7_/article/details/136383466

版权

A-B 数对

题目背景

出题是一件痛苦的事情！

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

题目描述

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

输入格式

输入共两行。

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

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

输出格式

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

样例 #1

样例输入 #1

4 1
1 1 2 3

样例输出 #1

提示

对于 $75\%$ 的数据， $\leq N \leq 2000$ 。

对于 $100\%$ 的数据， $\leq N \leq 2 \times 10^5$ ， $\leq a_i <2^{30}$ ， $\leq C < 2^{30}$ 。

2017/4/29 新添数据两组

思路

1.对于A-B=C这个式子而言，等号两边都有变量，移项成A-C=B
2.题目转换成：遍历数据，在数组中寻找B，使得B能满足==a[i] - C（二分）

注意点

对于a[i]，满足条件的B可能有很多个，我们就要找到第一个>=B的数的下标（left）和第一个>B的下标（right）,ans[i] = right - left - 1
之前错误想成，找第一个<B作left，和找第一个>B作right，错在，left可能超出边界，因为要找的B可能就是第一个数

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

typedef long long ll;

const int N = 200010;
int a[N];
int n,c;

int lower(int b)
{
	int l = 0,r = n - 1;
	while(l < r)
	{
		int mid = l + r >> 1;
		if(a[mid] >= b) r = mid;
		else l = mid + 1;
	}
	return l;
}

int upper(int b)
{
	int l = 0,r = n - 1;
	while(l < r)
	{
		int mid = l + r >> 1;
		if(a[mid] > b) r = mid;
		else l = mid + 1;
	}
	return l;
}

int main()
{
	scanf("%d %d",&n,&c);
	for(int i = 0;i < n;i ++) scanf("%d",&a[i]);
	sort(a,a + n);
	ll res = 0;
	for(int i = 0;i < n;i ++)
	{
		int b = a[i] - c;
		//printf("%d %d\n",upper(b),lower(b));
		res += (upper(b) - lower(b));
	}
	printf("%lld",res);
	return 0;
}