leetcode611.有效三角形的个数

题目

给定一个包含非负整数的数组 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

解答

• 遍历+双指针优化

class Solution {
public:
    int triangleNumber(vector<int>& nums) {
        int result = 0; // 初始化结果
        sort(nums.begin(), nums.end()); // 对数组排序
        int n = nums.size();
        for (int i = 0; i < n - 2; i++) {
            if (nums[i] == 0) continue; // 跳过为0的状态
            int k = i + 2; // 第二个指针的初始化位置
            for (int j = i + 1; j < n - 1; j++) {
                while (k < n && nums[i] + nums[j] > nums[k]) {
                    k++;
                }
                result += k - j - 1;// k - j - 1 表示的是满足条件的第三个边的数量
            }
        }
        return result; // 返回结果
    }
};