标签:二分 int 递增 mid 三元组 ++ include scanf
题目
分析
直接暴力\(O(N^3)\),显然超时。
将三个数组排序,遍历\(b\)数组,二分找到\(a\)中小于\(b[i]\)的个数\(A\),找到\(c\)中大于\(b[i]\)的个数\(C\),\(ans += A * C\)。
时间复杂度,排序\(O(N \log N)\),查找\(O(N \log N)\),总体\(O(N \log N)\)。
代码
自己写二分
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 100010;
int n;
int a[N], b[N], c[N];
LL ans;
int main()
{
scanf("%d", &n);
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
for (int i = 0; i < n; i++) scanf("%d", &b[i]);
for (int i = 0; i < n; i++) scanf("%d", &c[i]);
sort(a, a + n);
sort(b, b + n);
sort(c, c + n);
/* 二分:
* 1. 找边界
* 2. 求mid,写check函数
* 3. 若l = mid则 +1,r = mid不用 +1
*
* 需要注意的是,二分出来的点若在边界则需要特判。
*/
for (int i = 0; i < n; i++)
{
int A = 0, C = 0;
int l = 0, r = n - 1;
while (l < r)
{
int mid = l + r + 1 >> 1;
if (a[mid] < b[i]) l = mid;
else r = mid - 1;
}
if (a[l] < b[i]) A = l + 1;
l = 0, r = n - 1;
while (l < r)
{
int mid = l + r >> 1;
if (c[mid] > b[i]) r = mid;
else l = mid + 1;
}
if (c[l] > b[i]) C = n - l;
ans += (LL)A * C;
}
cout << ans << endl;
return 0;
}
lower_bound(), upper_bound();
标签:二分,int,递增,mid,三元组,++,include,scanf 来源: https://www.cnblogs.com/optimjie/p/12346122.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。