LeetCode-611. 有效三角形的个数

给定一个包含非负整数的数组，你的任务是统计其中可以组成三角形三条边的三元组个数。

示例 1:

输入: [2,2,3,4]
输出: 3
解释:
有效的组合是: 
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
注意:

数组长度不超过1000。
数组里整数的范围为 [0, 1000]。

讲一下这题的思路，什么是有效三角形，两边之和大于第三边，

两边之差小于第三边，任务是统计数组当中可以组成三角形三条边

的三元组个数，所以我们先排序数组，然后可以用两重循环先枚举出第一条边

，第二条边。为了防止重复，第一重循环遍历i ，第二重循环每次都从比 i大1的

地方开始枚举，然后每次算出 nums[i] + nums[j]的值，即第三条边third的大小（此时

大于nums[j] 的值而小于third的值都可以作为第三边，所以找到最终的一个k值

即nums[k] 刚好小于third的值，这个k值可使用二分之查找来查找）

#include <iostream>
#include <vector>
using namespace std;

class sortClass {
public:
	void qs(vector<int>& arr, int left, int right) {

		if (left >= right) {
			return;
		}

		int i = left;
		int j = right;
		int key = arr[i];
		while (i < j) {
			while (i < j && arr[j] >= key) {
				j--;
			}
			swap(arr[i], arr[j]);
			while (i < j && arr[i] <= key) {
				i++;
			}
			swap(arr[i], arr[j]);
		}
		arr[i] = key;
		qs(arr, left, i - 1);
		qs(arr, i + 1, right);
	}

	/* 二分查找 */
	int bSearch(vector<int>& nums, int left, int right, int third) {
		int k = left;
		while (left <= right) {
			int mid = (left + right) / 2;
			if (nums[mid] < third) {
				k = mid;
				left = mid + 1;
			}
			else {
				right = mid - 1;
			}
		}
		return k;
	}


private:
	void swap(int& a, int& b) {
		int temp = a;
		a = b;
		b = temp;
	}
};


class Solution {
public:
	int triangleNumber(vector<int>& nums) {
		int count = 0;

		/* 没有三条边 */
		if (nums.size() < 3) {
			return 0;
		}
		sortarr(nums);
		for (int i = 0; i < nums.size(); i++) {
			for (int j = i + 1; j < nums.size(); j++) {
				int third = nums[i] + nums[j];                   //两边之和大于第三边
				//cout << "i:" << i << "----val:" << nums[i] << endl;
				//cout << "j:" << j << "----val:" << nums[j] << endl;
				//cout << "third:" << third << endl;
				int findVal = Sc.bSearch(nums, j, nums.size() - 1, third);
				//cout << "true third:" << nums[findVal] << endl;
				count += findVal - j;
				//cout << "count:" << count << endl;
			}
		}
		return count;
	}
	void sortarr(vector<int>& nums) {
		Sc.qs(nums, 0, nums.size() - 1);
	}
private:
	sortClass Sc;
};

int main() {

	//vector<int> v = { 24,3,82,22,35,84,19 };
	//vector<int> v = { 2, 2, 3, 4 };
	//vector<int> v = { 7,0,0,0 };
	vector<int> v = { 61, 73, 62, 57, 46, 11, 33, 79, 79, 60 };
	Solution* ps = new Solution();
	ps->sortarr(v);
	for (int i = 0; i < v.size(); i++) {
		cout << "print:" << v[i] << endl;
	}

	cout << ps->triangleNumber(v) << endl;
	system("pause");
	return 0;
}