给定一个包含非负整数的数组 nums ,返回其中可以组成三角形三条边的三元组个数。
示例 1:
输入: nums = [2,2,3,4]
输出: 3
解释:有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
示例 2:
输入: nums = [4,2,3,4]
输出: 4
提示:
1 <= nums.length <= 1000
0 <= nums[i] <= 1000
先对nums进行排序,然后固定最短边,进行同向双指针:
cpp
class Solution {
public:
int triangleNumber(vector<int>& nums) {
if (nums.size() < 3) {
return 0;
}
sort(nums.begin(), nums.end());
int ans = 0;
for (int i = 0; i < nums.size() - 2; ++i) {
int left = i + 1;
int right = i + 2;
while (right < nums.size()) {
// 当两短边之和大于长边时
// 中边取[left, right - 1]之间的所有值都可以组成有效三角形
if (left == right || nums[i] + nums[left] > nums[right]) {
ans += right - left;
++right;
} else {
++left;
}
}
}
return ans;
}
};
如果nums的长度为n,则此算法时间复杂度为O(n2^22),空间复杂度为O(logn)。