题目描述
小红拿到了一个长度为 nnn 的数组,数组中的元素都是正整数。
小红想让你回答以下三个问题,取两个数乘积大于 kkk 的方案数、取两个数乘积等于 kkk 的方案数、取两个数乘积小于 kkk 的方案数。
注:两个数是不放回且同时取的。例如对于数组[1,2,3,4,5]而言,取[1,2]和[2,1]我们认为是同一种方案。
但是,如果有两个数相等,那么取相等但位置不同的数不认为是同一种取法。例如对于数组[2,2,2,3]而言,有三种方案可以取到[2,2]。
这个题一看到我就想到了二分,但是由于我确实不太熟,所以没有做。但是好像很多二分用的都是upper_bound和lower_bound,所以还是熟悉一下
大佬题解:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll a[300005];
int n;
ll k;
ll ans1,ans2,ans3,v1,v2;
inline ll read(){//当时为了优化我的O(n^2)常数用的read()
char ch = getchar();ll x = 0, f = 1;
while(ch<'0'||ch>'9'){if(ch == '-') f = -1;ch = getchar();}
while('0'<=ch && ch <= '9'){x = x*10+ch-'0';ch = getchar();}
return x*f;
}
int main()
{
n = read();
k = read();
for(int i = 1;i<=n;++i)
a[i] = read();
sort(a+1,a+1+n);
for(int i = 1;i<=n;++i)
{
int l = lower_bound(a + 1 + i,a + 1 + n,1.0 * k / a[i]) - a;
int r = upper_bound(a + 1 + i,a + 1 + n,1.0 * k / a[i]) - a;
ans1+=l-i-1;
ans3+=n-r+1;
ans2+=r-l;
}
printf("%lld %lld %lld",ans3,ans2,ans1);
return 0;
}