A-B 数对

题源 C已知,C+B=A,枚举所有的B,在数组中找A的个数,巧妙之处就是得用STL的upper_bound和lower_bound。不过也能用双指针

题目背景

出题是一件痛苦的事情!

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

题目描述

给出一串正整数数列以及一个正整数 C,要求计算出所有满足 A−B=C 的数对的个数(不同位置的数字一样的数对算不同的数对)。

输入格式

输入共两行。

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

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

输出格式

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

输入输出样例

输入 #1复制

4 1
1 1 2 3

输出 #1复制

3

说明/提示

对于 75% 的数据,1≤N≤2000。

对于 100% 的数据,1≤N≤2×10^5,0≤ai​<2^30,1≤C<2^30。

 upper_bound和lower_boun方法

#include<bits/stdc++.h>
using namespace std;
int n,c;
int a[1000009]={0};
long long ant=0;
int main()
{
	
	cin>>n>>c;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	sort(a+1,a+1+n);
	for(int i=1;i<=n;i++){
		ant+=upper_bound(a+1,a+1+n,a[i]+c)-lower_bound(a+1,a+1+n,a[i]+c);
	}
	cout<<ant;
	return 0;
 } 

 双指针

#include<bits/stdc++.h>
using namespace std;
int n,c;
int a[1000009]={0};
long long ant=0;
int main()
{
	
	cin>>n>>c;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	sort(a+1,a+1+n);
	int L=1,L1=1;
	for(int i=1;i<=n;i++){
		
		while(L<=n&&a[L]-a[i]<c){
			L++;
		} 
		while(L1<=n&&a[L1]-a[i]<=c){
			L1++;
		}
		if(a[i]+c==a[L1-1]&&a[i]+c==a[L]){
			ant+=L1-L;
		}
//		cout<<L<<" "<<L1<<" "<<ant<<endl;
	}
	
	cout<<ant;
	return 0;
 } 

  • 5
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值