P1102 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%75% 的数据,1 \leq N \leq 20001≤N≤2000。
对于 100%100% 的数据,1 \leq N \leq 2 \times 10^51≤N≤2×105。
保证所有输入数据都在 3232 位带符号整数范围内。
方法一
使用自己实现的lower_bound和upper_bound函数
#include<iostream>
#include<algorithm>
using namespace std;
long long int N, C;
long long int p[10000000];
long long int Lower_bound(long long int A[], long long int left, long long int right, long long int x)//返回第一个等于x的位置
{
long long int mid, ans = 0;
while (left <= right)
{
mid = left + (right - left) / 2;
if (A[mid] >= x)
right = (ans = mid) - 1;
else
left = mid + 1;
}
if (A[ans] == x)
return ans;
else
return -1;
}
long long int Upper_bound(long long int A[], long long int left, long long int right, long long int x)//返回第一个大于x的位置
{
long long int mid, ans = 0;
long long int Asize = right;
while (left < right)
{
mid = left + (right - left) / 2;
if (A[mid] > x)
right = (ans = mid);
else
left = mid + 1;
}
if (A[ans] > x)
return ans;
else
return Asize;
}
int main()
{
cin >> N >> C;
for (long long int i = 0; i < N; i++)
cin >> p[i];
sort(p, p + N);
long long int sum = 0;
for (long long int i = 0; i < N - 1; i++)
if (Lower_bound(p, 0, N, p[i] + C) != -1)
sum += Upper_bound(p, 0, N, p[i] + C) - Lower_bound(p, 0, N, p[i] + C);
cout << sum << endl;
}
方法二
使用STL中提供的lower_bound和upper_bound函数,注意返回值是指针
#include<iostream>
#include<algorithm>
using namespace std;
long long int N, C;
long long int p[10000000];
int main()
{
cin >> N >> C;
for (long long int i = 0; i < N; i++)
cin >> p[i];
sort(p, p + N);
long long int sum = 0;
for (long long int i = 0; i < N - 1; i++)
if (lower_bound(p, p + N, p[i] + C) - p != N)
sum += upper_bound(p, p + N, p[i] + C) - lower_bound(p, p + N, p[i] + C);
cout << sum << endl;
}