给定一个包含非负整数的数组,你的任务是统计其中可以组成三角形三条边的三元组个数。
示例 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;
}