leetcode(40)_611_medium_有效三角形的个数_python

有效三角形的个数

题目描述：
给定一个包含非负整数的数组，你的任务是统计其中可以组成三角形三条边的三元组个数。
示例 :
输入: [2,2,3,4]
输出: 3
解释:
有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
提示：

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

解法

1. 本题与 三数之和 较为相似，最直接的思路都是排序后，进行三重循环来统计可行解，我们用下标(i, j, k)来表示某一可行解，其中i < j < k。
2. 对于三数之和问题，我们在确定了一组解(i, j, k)后，要找与(i, j + 1)匹配的k^*，不必再遍历一次，因为nums[i]不变，nums[j] <= nums[j + 1]，所以必然有nums[k^*] <= nums[k]，因此只需要反向遍历一次 k 即可，这样就形成了j, k双指针相向遍历的情形，将复杂度降到了O(n²)。
3. 类似的，本题也可以使用双指针进行优化。我们在确定了(i, j)后，只需要找到满足nums[i] + nums[j] > nums[k]的 k 的最大值k_i,j，在该最大值的左侧必然也都是可行解；在我们要找与(i, j + 1)匹配的 k 时，同样可以找 k 的最大值 k_i,j+1，显然有k_i,j+1 >= k_i,j，所以我们只需要正向遍历一次 k 即可，形成了j, k双指针同向遍历的情形，复杂度降到了O(n²)。

代码

class Solution:
    def triangleNumber(self, nums: List[int]) -> int:
        nums.sort()
        res = 0
        n = len(nums)
        for i in range(n - 2):
            if nums[i] == 0:
                continue
            k = i + 2
            for j in range(i + 1, n - 1):
                while k < n and nums[i] + nums[j] > nums[k]:
                    k += 1
                res += k - j - 1
        return res
   

测试结果

执行用时：1548 ms, 在所有 Python3 提交中击败了 39.61% 的用户
内存消耗：14.8 MB, 在所有 Python3 提交中击败了 90.81% 的用户

说明

算法题来源：力扣(LeetCode)