P1102 A-B数对 （二分查找）

题目链接：P1102

解题思路：

要找A-B=C，变成A=B+C，将数排序，遍历每一个数，用二分查找查看是否存在B+C，因为不同位置数字算不同一对，所以要特殊处理，开数组记录的话可能会MLE，所以用map来存数字出现次数

AC代码：

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <map>
#define ll long long
using namespace std;
ll n,c,mx,cnt,a[300010];
inline ll read() // 快读
{
    ll x = 0,y = 1; char c = getchar();
    while(c < '0' || c > '9') {if(c == '-') y = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x*10+c-'0', c = getchar();
    return x*y;
}
bool bfind(ll x) // 二分
{
    int l = 0, r = n-1;
    int mid = (r-l)/2+l;
    while(true)
    {
        if(l == r)
        {
            if(a[l] == x) return true;
            else return false;
        }
        if(a[mid] >= x)
        {
            r = mid;
            mid = (r-l)/2+l;
        } else if(a[mid] < x) {
            l = mid+1;
            mid = (r-l)/2+l;
        }
    }
}
int main()
{
    map<ll, ll> m;
    cnt = 0;
    n = read();
    c = read();
    for(int i = 0; i < n; i++)
    {
        a[i] = read();
        m[a[i]]++; // 记录出现次数
    }
    sort(a,a+n);
    mx = a[n-1]; // 记录一下最大值
    for(int i = 0; i < n; i++)
    {
        if(a[i]+c > mx) continue; // 如果超过最大值，就没必要去查找了
        else if(bfind(a[i]+c))
            cnt += 1*m[a[i]+c]; // 算不同位置的对数
    }
    printf("%lld",cnt);
    return 0;
}