给定数组求有效三角形的三元组个数

前言

题目要求：给定一个包含非负整数的数组 nums ，返回其中可以组成三角形三条边的三元组个数。

思路：我们知道三角形满足任意2条边之和大于第三边，之差小于第三边即可，这里我们将数组按升序排好，进行2个for循环遍历，第一个循环为第一条边i（0，length-1），第二个循环为第二条边j（第一条边+1，length-2）因为从左往右遍历第一条边到倒数第三个数结束，而第二条边到倒数第二个数就可以结束。对于第三条边k（k>j）我们从右往左遍历，直到遇到能满足i+j>k的情况那么则res加上k-j，并继续j的循环，k重新回到最右边位置，因为j-k之间的数都能满足i+j>k。对于俩边之差i-j一定是满足i-j<k的因为初始数组升序排序且k最大。

代码如下（python）：

class Solution:
    def triangleNumber(self,nums):
        nums.sort()
        length = len(nums)
        res= 0
        for i in range(0,length-2):
            for j in range(i+1,length-1):
                k = length - 1
                while j < k:
                    if nums[i] + nums[j] > nums[k]:
                        res += k-j
                        break
                    k -= 1
        return res

a = Solution()
nums = list(map(int,input().split()))
print(a.triangleNumber(nums))

2.示例