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

题目：
给定一个包含非负整数的数组，你的任务是统计其中可以组成三角形三条边的三元组个数。

思路：
先对数组进行排序，然后循环取三角形的三条边。
枚举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