解法:
先将数组排序,然后确定其中两个边,通过双指针法或二分查找找到第三个边满足条件。具体来说,用指针i从左往右遍历数组,对于每个i,在i+1到n-1的范围内使用两个指针left和right,寻找满足nums[left]+nums[right]>nums[i]的最大的left。此时,在left到right之间的任意一个位置j,三个数nums[i]、nums[left]、nums[j]也能构成三角形。我们将left移动到left+1,继续寻找下一个满足条件的left。
那么如何确定right的移动?由于我们要找最大的left,若right能满足条件,那么right+1必然也满足。同时,若right不能满足条件,那么right右侧的数必然也不能满足条件,因此我们可以直接跳出right的搜索,开始下一个i的搜索。
时间复杂度
排序时间复杂度为O(nlogn),对于每个i,left和right指针最多分别移动n-1次,因此总时间复杂度为O(nlogn+n^3)=O(n^3)。
空间复杂度
排序需要O(logn)的空间,双指针算法需要常数级别的空间,因此空间复杂度为O(logn)。
C++ 代码
class Solution {
public:
int triangleNumber(vector<int>& nums) {
sort(nums.begin(), nums.end()); // 先排序
int n = nums.size(), result = 0;
for (int i = 0; i < n - 2; i++) { // 枚举第一条边
int k = i + 2; // 初始化第三条边的位置
for (int j = i + 1; j < n - 1 && nums[i] != 0; j++) { // 枚举第二条边
while (k < n && nums[i] + nums[j] > nums[k]) {
k++; // 右移第三条边
}
result += k - j - 1; // 第三条边必须在[k, n-1]中,且小于nums[i]+nums[j]
}
}
return result;
}
};
力扣611. 有效三角形的个数
最新推荐文章于 2024-07-13 13:36:16 发布