CF-Round#627-div3-D题
D. Pair of Topics
二分排序题。
题目大意:要你找一下符合题目要求定义的pair。给了一个序列是老师喜欢某个话题的程度。再给了一个序列是同学喜欢某个话题的程度。选择两个话题:话题i和话题j,如果老师对这两个话题喜欢的程度之和大于同学对这两个话题喜欢的程度之和。那么这一组pair称为好pair。
输出好pair的数量。
先开始我以为是一个组合问题。后来仔细想了下。似乎组合解决不了。
为了不超时。
数据量达到了2e5.数据范围达到了1e9。用ll没得说。然后不要超时就行。
不超时的话用二分upper_bound(好久没用差不多又忘了。还是要好好复习的,三省)
很容易想到求差。sort一下。然后记录0的个数和大于0的个数。把小于0的另存。
然后遍历小于0的。
找到第一个比这个小于0的相反数大的第一个索引。后面的索引都可以与这个小于0的数进行组合。维护ans就行。这个地方用二分啦
然后大于0的和等于0的可以组合。大于0的和大于0的可以组合。把他们加上就行
代码部分:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;
int n;
ll a[N];
ll b[N];
ll c[N];
ll sum0, sum1;
ll sum;
vector<ll> v;
int main()
{
cin >> n;
for (int i = 0; i < n; i++)
{
scanf ("%I64d", &a[i]);
}
for (int i = 0; i < n; i++)
{
scanf ("%I64d", &b[i]);
c[i] = a[i] - b[i];
if (c[i] > 0)
{
sum1++;
}
else if (!c[i])
{
sum0++;
}
else
{
v.push_back(c[i]);
}
}
sort(c, c + n);
ll ans = 0;
int k = v.size();
for (int i = 0; i < k; i++)
{
ans += (n - (upper_bound(c, c + n, -v[i]) - c));
}
ans += (sum0 * sum1 + sum1 * (sum1 - 1) / 2);
cout << ans << endl;
return 0;
}