P1102 A-B 数对
1.题目
题目链接
2.分析
首先先排序(为了后续二分查找方便)
依次对每个元素A,运用二分找到>=A+C的最小边界和>A+C的最小边界,两者之差便是所求的实数对个数
3.代码
1.ans开int (❌)
#include <iostream>
#include <algorithm>
using namespace std;
int Search(int a[], int n, int flag) //找到》= 和 >的最小值
{
int ans = 0;
for (int i = 0; i < n; i++)
{
int index = a[i] + flag;
int t0 = 0, t1 = 0;
int l = 0, r = n - 1;
while (l <= r)
{
int mid = l + ((r - l) >> 1);
if (a[mid] >= index)
{
t0 = mid;
r = mid - 1;
}
else
{
l = mid + 1;
}
}
if (!t0) //有可能找不到符合条件的
t0 = n;
l = 0, r = n - 1; //重新赋值
while(l <= r) //找>index的最小下标
{
int mid = l + ((r - l) >> 1);
if (a[mid] > index)
{
t1 = mid;
r = mid - 1;
}
else
{
l = mid + 1;
}
}
if (!t1)
t1= n;
ans += t1 - t0;
}
return ans;
}
int main()
{
int n, flag;
cin >> n >> flag;
int* a = new int[n];
for (int i = 0; i < n; i++)
cin >> a[i];
sort(a, a + n);
cout << Search(a,n, flag);
delete[]a;
return 0;
}
2.ans开long long (√)
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
ll Search(int a[], int n, int flag) //找到》= 和 >的最小值
{
ll ans = 0;
for (int i = 0; i < n; i++)
{
int index = a[i] + flag;
int t0 = 0, t1 = 0;
int l = 0, r = n - 1;
while (l <= r)
{
int mid = l + ((r - l) >> 1);
if (a[mid] >= index)
{
t0 = mid;
r = mid - 1;
}
else
{
l = mid + 1;
}
}
if (!t0) //有可能找不到符合条件的
t0 = n;
l = 0, r = n - 1; //重新赋值
while(l <= r) //找>index的最小下标
{
int mid = l + ((r - l) >> 1);
if (a[mid] > index)
{
t1 = mid;
r = mid - 1;
}
else
{
l = mid + 1;
}
}
if (!t1)
t1= n;
ans += t1 - t0;
}
return ans;
}
int main()
{
int n, flag;
cin >> n >> flag;
int* a = new int[n];
for (int i = 0; i < n; i++)
cin >> a[i];
sort(a, a + n);
cout << Search(a,n, flag);
delete[]a;
return 0;
}
3.优化,利用upper_bound(),lower_bound()
#include <iostream>
#include <algorithm> //upper_bound lower_bound的头文件
using namespace std;
typedef long long ll;
int main()
{
ll ans = 0;
int n, flag;
cin >> n >> flag;
int* a = new int[n];
for (int i = 0; i < n; i++)
cin >> a[i];
sort(a, a + n); //必须排序,为了后面的二分查找
for (int i = 0; i < n; i++)
ans += ((upper_bound(a, a + n, a[i] + flag) - a) - (lower_bound(a, a + n, a[i] + flag) - a));
cout << ans;
delete[]a;
return 0;
}
4.总结
通过此题加深了我对二分的理解,增强对数据的感知能力(开 long long)
比如:
while(r<=l) // =不要省掉!!
找到>= <某个数的下标 的方法
学到了upper_bound lower_bound函数
其他方法,后续学到了会重新补上
5.更新日志
2022.5.12 整理上传
欢迎交流、讨论、指正~
不正确、不理解之处欢迎评论留言~