洛谷——A - B数对

题目描述

出题是一件痛苦的事情！

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

好吧，题目是这样的：给出一串数以及一个数字 CC，要求计算出所有 A - B = CA−B=C 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

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

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

输出格式

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

输入输出样例

输入 #1复制

4 1
1 1 2 3

输出 #1复制

3

说明/提示

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

对于 100% 的数据，1≤N≤2×105。

保证所有输入数据绝对值小于 2^{30}C≥1。

2017/4/29 新添数据两组

解题思路：直接二分， 具体看代码。

Code:

#include <iostream>
#include <stdio.h>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 52e5 + 10;

LL n, b, ans;
LL a[N], c, s;

signed main()
{
    cin >> n >> c;
    for(int i = 0; i < n; i ++ )  cin >> a[i];
    sort(a, a + n);
    for(int i = 0; i < n; i ++ )
    {
        s = a[i] + c;
        int l = 0, r = n - 1;
        while(l < r)
        {
            int mid = l + r >> 1;
            if(a[mid] >= s)  r = mid;
            else  l = mid + 1;
        }
        if(a[l] != s)  continue;
        else
        {
            b = l;
            l = 0, r = n - 1;
            while(l < r)
            {
                int mid = l + r + 1 >> 1;
                if(a[mid] <= s)  l = mid;
                else r = mid - 1;
            }
            ans += r - b + 1;
        }
    }
    cout << ans << endl;
}