leetcode每日一题—611.有效三角形的个数

最新推荐文章于 2025-09-26 15:20:15 发布

原创最新推荐文章于 2025-09-26 15:20:15 发布 · 574 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

Python 同时被 2 个专栏收录

484 篇文章

订阅专栏

leetcode

461 篇文章

订阅专栏

该博客介绍了如何使用排序和二分查找算法来解决寻找数组中能组成三角形的三元组问题。首先对数组排序，然后通过两个指针遍历数组，使用二分查找找到可能的第三个边长，确保构成三角形的条件满足。这种方法提高了搜索效率，减少了不必要的计算。

题目：
给定一个包含非负整数的数组，你的任务是统计其中可以组成三角形三条边的三元组个数。
在这里插入图片描述
思路：
先对数组进行排序，然后循环取三角形的三条边。
枚举nums[i],nums[j]作为较小的两条边，然后利用二分法查找符合条件的最大的nums[k]（其中i,j,k符合条件i<j<k，且nums[i]<=nums[j]<=nums[k])。
构成三角形的条件是：两边之和大于第三边。
因为数组有序，所以仅需满足nums[i]+nums[j]>nums[k]即可满足构成三角形的条件。

解答：

class Solution:
    def triangleNumber(self, nums: List[int]) -> int:
        nums.sort()
        n=len(nums)
        res=0
        for i in range(n):
            a=nums[i]
            for j in range(i+1,n):
                b=nums[j]
                left,right=j+1,n-1
                #二分查找满足nums[k]<nums[i]+nums[j]的最大下标k
                end=-1
                while left<=right:
                    mid=(left+right)//2
                    #print(left,right,mid,nums[mid],a,b)
                    if nums[mid]<a+b:
                        left=mid+1
                        end=mid
                    else:
                        right=mid-1
                if end!=-1:
                    res+=end-j
                #print(i,j,end,res)
        return res