二分查找的变式

题目描述

小红拿到了一个长度为 n 的数组，数组中的元素都是正整数。
小红想让你回答以下三个问题，取两个数乘积大于 kkk 的方案数、取两个数乘积等于 kkk 的方案数、取两个数乘积小于 kkk 的方案数。
注：两个数是不放回且同时取的。例如对于数组[1,2,3,4,5]而言，取[1,2]和[2,1]我们认为是同一种方案。
但是，如果有两个数相等，那么取相等但位置不同的数不认为是同一种取法。例如对于数组[2,2,2,3]而言，有三种方案可以取到[2,2]。

思路

将数组a从小到大进行排序后，从数组的第一个位置遍历至倒数第二个，第i次遍历，固定a[i]为要取的两个的其中一个，再通过二分查找找到另一个数a[m]，该数为所有与a[i]乘积小于k中最大的，则此次遍历可得到乘积小于k的个数为m-i，最后全部相加，即使所有乘积小于k的个数。

查找乘积大于k的个数的思路与上面类似。

//返回的索引为所有比要查找的值小的最大值
long search1(long x,long l,long r){
	long m;
	while(l<=r){
		m=(l+r)/2;
		if(x*a[m]<k)
			l=m+1;
		else
            r=m-1;
	}
	return r;
}
//返回的索引为所有比要查找的值大的最小值
long search2(long x,long l,long r){
	long m;
	while(l<=r){
		m=(l+r)/2;
		if(x*a[m]>k)
			r=m-1;
		else
            l=m+1;
	}
	return l;
}