leetcode 有效三角形的个数

题目

给定一个包含非负整数的数组，你的任务是统计其中可以组成三角形三条边的三元组个数。

示例 1:

输入: [2,2,3,4]
输出: 3
解释:
有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
注意:

数组长度不超过1000。
数组里整数的范围为 [0, 1000]。

分析

1.此题类似于求三数之和为零。
2.三边构成三角形的条件，任意两边之和大于第三边，任意两边之差小于第三边。

想法一

与三数之和一样，先确定两条边，再求第三条边。
1.由于数组里整数的范围为[0, 1000]，故先对输入进行判定：若数组中大于0的个数若小于3个，则直接return 0；否则去除数组中的零，再执行后续操作。
2.对数组进行排序，这样除了方便步骤1中的数非零数的个数和去零外，还能是确定第三条边不可缺的条件。
3.先确定两个指针i，j，完成双循环，其中j的初始条件为j = i + 1；
4.确定第三条边的范围：设立左坐标index1，初始化为0，每次循环+1直至nums[index1]>nums[j]-nums[i]，右坐标index2，初始化为nums.size()-1，每次循环-1直至nums[index1]<nums[j]+ nums[i]。
5.找到左右坐标后分别于i与j比对，若index1>i，index1–；若index2<j，index2++，最终输出res += index2-index1 -1。

代码一

class Solution {
public:
    int triangleNumber(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        while(nums.size() && 0 == nums[0])
            nums.erase(nums.begin());
        int len = nums.size();
        if(len < 3) return 0;  
        int res = 0;
        for(int i = 0; i < len - 1; i++)
        {
            for(int j = i+1; j < len; j++)
            {
                int index1 = 0,index2 = len-1;
                while(nums[index1] <= nums[j] - nums[i]) index1++;
                if(index1 > i) index1--;
                while(nums[index2] >= nums[j] + nums[i]) index2--;
                if(index2 < j) index2++;
                res += (index2 - index1 -1);
            }
        }
        return res/3;
    }
};

想法二

该想法来自大佬LuissssGao感觉比我的好不少！！！
每次先拿到第三条边，然后找前两条边的范围！这样复杂度可以下降到0（n^2）!代码如下（C++）

class Solution {
public:
    int triangleNumber(vector<int>& nums) {
        int count = 0, size = nums.size();
        sort(nums.begin(), nums.end());
        for (int i = size - 1; i >= 2; i--) {    // 先拿到第3条边
            int left = 0, right = i - 1;    // 前2条边
            while(left < right) {
                if (nums[left] + nums[right] > nums[i]) {
                    count += (right - left);    // 找到区间以后，就更新第2条边，构建新组合，并记录上个组合的总数
                    right--;
                }
                else {
                    left++;    // 调整第1条边
                }
            }
        }
        return count;
    }
};