【每日刷题】P1102 A-B 数对题解

【每日刷题】P1102 A-B 数对题解

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

3

提示

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

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

2017/4/29 新添数据两组

c解法：双指针

如下图，左右指针维护的是一段连续的值为B+C=A的区间，然后这个长度就是题目的方案数。

#include<stdio.h>
#include<math.h>


int a[200005];
int n, c;



int cmp_int(const void* e1, const void* e2) {
	return *(int*)e1 - *(int*)e2;
}

int main() {
	int  numC;
	long long ans = 0;
	scanf("%d %d", &n, &c);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
	}
	qsort(a + 1, n, sizeof(a[1]), cmp_int);
	int l=1,r=1;
	for(int i=1;i<=n;i++){
		numC=a[i]+c;
		while(l<=n&&a[l]<numC)l++;//把左指针移到这个区间的最左边
		while(r<=n&&a[r]<=numC)r++;//把右指针移到这个区间的最右边+1
		ans+=r-l;
	}
	printf("%lld\n",ans);
	return 0;
}

c解法：二分

二分其实也是找到一段连续的值为A=B+C的区间，所以我们用二分的方法找到左右区间的临界值

#include<stdio.h>
#include<math.h>


int a[200005];
int n, c;

int findlen1(int x) {//找到数组中为x的数的最小下标，若未找到则返回0
	int l=1,r=n;
	while (l <= r) {
		int mid = (l + r) / 2;
		if (a[mid] <x) {//这里，没有等号，相当于找到这个区间的最左边-1
			l = mid + 1;
		} else {
			r = mid - 1;
		}
	}
	if(a[l]==x)return l;
	else return 0;
}
int findlen2(int x) {
	int l=1,r=n;
	while (l <= r) {
		int mid = (l + r) / 2;
		if (a[mid] <=x) {//这里有等号，就是说找到这个区间的最右边
			l = mid + 1;
		} else {
			r = mid - 1;
		}
	}
	if(a[r]==x)return r;
	else return -1;//没找到就说明没有这个区间，-1是因为下面在求区间长度的时候会有+1
}

int cmp_int(const void* e1, const void* e2)
{
    return *(int*)e1 - *(int*)e2;
}

int main() {
	int  numC;
	long long  ans = 0;
	scanf("%d %d", &n, &c);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
	}
	qsort(a+1, n, sizeof(a[1]), cmp_int);
	for (int i = 1; i <= n; i++) {
		numC = a[i] + c;
		ans += (findlen2(numC)-findlen1(numC))+1;
	}
	printf("%lld\n", ans);
}

c++写法：map

用STL的map函数去映射，主要是因为直接用数组去的话映射范围会爆掉。

#include<stdio.h>
#include<math.h>
#include<map>
using namespace std;


int a[200005];
map<long,long>num;
int main() {
	int n, c;
	long long ans=0;
	scanf("%d %d", &n, &c);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
		num[a[i]]++;//统计每个数字出现的个数
	}
	for (int i = 1; i <= n; i++) {
		if(num[a[i]+c]!=0){
			ans+=num[a[i]+c];
		}
	}
	printf("%lld\n",ans);
}