P1102 A-B 数对

合集 - 刷题(27)

1.uva253骰子涂色2023-11-302.洛谷uva2202023-11-303.uva12096集合栈计算机 The SetStack Computer2023-12-054.uva101The Blocks Problem2023-12-055.uva400 Unix Is命令 Unix ls2023-12-106.uva10391 复合词 Compound Words2023-12-107.小球下落 Dropping Balls （uva679）2023-12-198.树的层次遍历 Trees on the level uva1222023-12-199.树 Tree uva5482023-12-1910.天平 Not so Mobile uva8392023-12-19

11.P1102 A-B 数对2024-02-10

12.成最多水的容器2024-02-1213.P3367 【模板】并查集2024-02-1314.947. 移除最多的同行或同列石头2024-02-1415.P1082 [NOIP2012 提高组] 同余方程2024-02-2216.803 打砖块2024-02-2317.B3644 【模板】拓扑排序 / 家谱树2024-02-2318.U107394 拓扑排序模板2024-02-2319.P1137 旅行计划2024-02-2420.936 戳印序列2024-02-2721.P4017 最大食物链计数2024-02-2722.2050 并行课程 III2024-02-2723.P3366 【模板】最小生成树2024-03-0124.P2330 [SCOI2005] 繁忙的都市2024-03-0425.Abbott的复仇 Abbott's Revenge2024-03-0626.P1960 郁闷的记者2024-03-2227.P4391 [BOI2009] Radio Transmission 无线传输2024-05-03

收起

原题链接

解法一：二分搜素

首先我们知晓A-B=C,那么A=B+C,我们只需要遍历数组中的每一个元素然后在数组中搜素a[i]+c的值是否存在即可。

Code

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
ll a[N];
int main(){
    int n,c;
    ll sum=0;
    cin>>n>>c;
    map<ll,int> map1;
    for (int i=1;i<=n;i++){
        cin>>a[i];
        map1[a[i]]++;
    }
    for (int i=1;i<=n;i++)
        if (map1.count(c+a[i])) sum+=map1[c+a[i]];
        cout<<sum<<endl;
}

 

解法二：双指针

我们先将数组按照非降序排列，我们用r1表示a[r1]-a[i]<c;r2表示a[r2]-a[i]<=c;那么对于每个a[i]我们只在需要sum+=r2-r1即可

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int a[N];
int main(){
    int n,c;
    ll sum=0;
    cin>>n>>c;
    for (int i=1;i<=n;i++) cin>>a[i];
    sort(a+1,a+n+1);
    int r1=1,r2=1;
    for (int i=1;i<=n;i++){
        while (r1<=n && a[r1]-a[i]<c) r1++;
        while (r2<=n && a[r2]-a[i]<=c) r2++;
        sum+=r2-r1;
    }
    cout<<sum<<endl;
}

Ps：如果a数组本身就是非降序的话，法二时间复杂度只有O（n）