611. 有效三角形的个数
原始题目链接:https://leetcode-cn.com/problems/valid-triangle-number/
给定一个包含非负整数的数组,你的任务是统计其中可以组成三角形三条边的三元组个数。
示例 1:
输入: [2,2,3,4]
输出: 3
解释:
有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
注意:
数组长度不超过1000。
数组里整数的范围为 [0, 1000]。
解题思路:
判断三条边能组成三角形的条件为:任意两边之和大于第三边,任意两边之差小于第三边。所以三条边长从小到大为 a、b、c,当且仅当 a + b >
c 这三条边能组成三角形。所以先给数组排序,然后注意,使用倒叙的方式来计算满足不等式的条件,这样计算比较快。具体实现看代码和注释。
代码实现:
class Solution:
def triangleNumber(self, nums: List[int]) -> int:
# 排序数组
nums.sort()
# 构成三角形的条件:a <= b <= c,只需要保证 a + b > c
res = 0
# 这里用倒叙计算
for i in range(len(nums) - 1, 1, -1):
l, r = 0, i - 1
while l < r:
# 只要nums[l] + nums[r] > nums[i]
# 那么l+1 l+2 ...... l+r-1的索引都满足条件的个数一共r - l个
if nums[l] + nums[r] > nums[i]:
res = res + r - l
r = r - 1
else:
l = l + 1
return res
参考文献:
https://leetcode-cn.com/problems/valid-triangle-number/solution/ming-que-tiao-jian-jin-xing-qiu-jie-by-jerring/