这道题跟之前做过的 3Sum 很相似(先排序)。O(n^3)的做法肯定超时,应该先固定一个值,然后用two pointers遍历,这样达到O(n^2)的复杂度。不过这里遇到的问题是,如果你固定最小的值,当右边的left和right构成的三角行,此时不合法,应该怎么处理?这时既可以左移right,又可以右移left,但这两种情况都要考虑,因为这种操作都可能使三角形合法。这样就很难处理。
答案中有简单又很启发人的办法,就是你固定最右边的最大值作为一条边,然后取0为left,最大值左边的第一个为right,此时如果不合法,即我们需要增大left和right处的值的和,因为right已经到最右了,所以只能让left右移;如果合法,此时,left到right之间的位置都是left可取的,计数之后,right右移,再循环判断即可。
class Solution {
public:
int triangleNumber(vector<int>& nums) {
if(nums.size() < 3)return 0;
sort(nums.begin(), nums.end());
int count = 0;
for(int i = nums.size() - 1; i >= 2; i--){
int left = 0, right = i - 1;
while(left < right){
if(nums[left] + nums[right] > nums[i])//合法
{
count += right - left;
right--;
}
else left++; //不合法
}
}
return count;
}
};