【LeetCode】611. Valid Triangle Number 解题报告（Python）

原创

负雪明烛 2021-07-14 13:33:42 ©著作权

文章标签 LeetCode 算法 leetcode 数据结构排序 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者负雪明烛的原创作品，请联系作者获取转载授权，否则将追究法律责任

作者：负雪明烛
id： fuxuemingzhu
个人博客： http://fuxuemingzhu.cn/

题目地址：https://leetcode.com/problems/valid-triangle-number/description/

题目描述：

Given an array consists of non-negative integers, your task is to count the number of triplets chosen from the array that can make triangles if we take them as side lengths of a triangle.

Example 1:

Input: [2,2,3,4]
Output: 3
Explanation:
Valid combinations are: 
2,3,4 (using the first 2)
2,3,4 (using the second 2)
2,2,3

Note:

The length of the given array won’t exceed 1000.
The integers in the given array are in the range of [0, 1000].

题目大意

给出了一个数组，求在这个数组中能组成多少个三角形。

解题方法

看了下，这个规模是 1000，那么用 O(n^3) 的暴力解法肯定就TLE了。那么只能用更高效的算法。

这个方法是这样的，我们先对这个数组进行排序，排序之后我们来指出这个三角形中最长的边，那么剩下两条边的要求是：两边之和大于最长边。

所以，我们只需要对最长边进行变遍历，对两个短边再遍历即可。短边的遍历方法是：一个从 0 开始的短边 l ，一个从比最长边的短的一个边 r 开始向中间遍历。如果能够成三角形，那么说明比l长的短边和r结合也都能组成三角形。如果不能组成三角形，那么说明l太小了，需要向中间移动。

排序算法时间复杂度是 O(nlogn)，for循环时间复杂度为O(n)，while循环虽然有两个变量，但是这两个变量是同时向中间移动的关系，所以时间复杂度不会超过O(n)。总体的时间复杂度是O(n^2)。空间复杂度是O(1)。

注意，sorted() 函数不是原地排序，如果原地排序需要使用 nums.sort()。这就是我第一次提交失败的地方。

代码如下：

class Solution:
    def triangleNumber(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        nums.sort()
        count = 0
        for i in range(len(nums) - 1, 1, -1):
            l, r = 0, i - 1
            while l < r:
                if nums[l] + nums[r] > nums[i]:
                    count += r - l
                    r -= 1
                else:
                    l += 1
        return count