Valid Triangle Number

Valid Triangle Number

题目描述:

给定一个数组,要求找到数组中所有组合能够构成三角形的边的组合的个数。

例子:

具体描述看LeetCode611

解题思路:

一开始的思路是想要用回溯的方式找到所有的组合然后判断每个组合的合法性来计算所有的组合;但是TLE。后面发现其实我们可以将数组排序后利用排序的性质来找到所有合法的三角形边。主要的方法是外面设置两个循环找到三角形的两条边,寻找第三条边的时候,我们从第二条边的下一个数开始遍历,找到第一个大于等于前两条边和的位置和第二条边的下一个数的位置之间的所有数都满足构成三角形的情况。

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution {
public:
int triangleNumber(vector<int>& nums) {
int res = 0;
if (nums.size() < 3) return res;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size() - 2; i++){
if (nums[i] == 0) continue;
for (int j = i + 1; j < nums.size() - 1; j++){
int temp = nums[i] + nums[j];
int flag = j + 1;
while(flag < nums.size() && nums[flag] < temp){
flag++;
}//找到大于等于前两条边和的第一个位置。
res += (flag - (j + 1));
}
}
return res;
}
};