(ACM)三数较小之和

题目描述:

给定一个包含n(n <= 2000)个整数的无序数组,在数组中选三个数使得它们的和小于一个特殊的数,每个数字只能使用一次

请实现three_sum_smaller函数,传入的nums是数组,length是元素的个数,target表示选出来的三个数的和需要小于target。返回一个整数,表示返回方法的种类数

样例输入:

10 35
19 28 23 1 30 15 7 15 21 5

样例输出:

17

解题思路:暴力搜索+二分查找的联合使用,时间复杂度:n*n*log2n

首先对数组进行排序,然后通过暴力搜索的方法,选取第一个数num1(左边)和第二个数num2(中间),得到要查找的第三个数

Sum = Target - num1 - num2

然后我们通过二分查找法,查找第一个大于等于sum的数(lower_bound方法)

得到下标后,num3的下标减去num2的下标+1就是所需的结果

部分实现代码如下:

int lower_bound(vector<int> &nums, int left, int right, int x) {   //不存在则返回最后一个元素
	int mid;
	while (left < right) {
		mid = (left + right) / 2;
		if (nums[mid] >= x) {
			right = mid;
		}
		else {
			left = mid + 1;
		}
	}
	
	return left;
}

int three_sum_smaller(vector<int> &nums, int length, int target) {
	quick_sort(nums, 0, length-1);  //先排个序
	int count = 0;
	for (int i = 0; i < length; i++) {
		for (int j = i + 1; j < length; ++j) {
			int sum = target - nums[i] - nums[j];
			int k = lower_bound(nums, j + 1, length-1, sum);
			count += k - i;
		}
	}
	return count;
}