题目地址(611. 有效三角形的个数)

题目地址(611. 有效三角形的个数)

https://leetcode.cn/problems/valid-triangle-number/

题目描述

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

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 Code:

class Solution:
    def triangleNumber(self, nums: List[int]) -> int:
        # 三角形：任意两条边大于第三条边
        n = len(nums)
        nums.sort()
        ans = 0
        for i in range(n):
            for j in range(i+1,n):
                left,right,k = j+1,n-1,j
                while left<=right:
                    mid = (left+right)//2
                    if nums[mid] < nums[i] + nums[j]:
                        k = mid # 满足k所在的地方一定是最先满足条件的地方
                        left = mid+1
                    else:
                        right = mid-1
                ans += k-j
        return ans 

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n^2)$
• 空间复杂度：$O(logn)$

代码

• 语言支持：Python3

Python3 Code:

class Solution:
    def triangleNumber(self, nums: List[int]) -> int:
        # 三角形：任意两条边大于第三条边
        n = len(nums)
        nums.sort()
        ans = 0
        for i in range(n):
            k = i
            for j in range(i+1,n):
                while k+1<n and nums[k+1]<nums[i]+nums[j]: # k指向能够满足条件的最后一个位置
                    k += 1 
                ans += max(k-j,0)
        return ans 

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n^2)$
• 空间复杂度：$O(logn)$