diff options
author | Jack Sangdahl <[email protected]> | 2025-09-25 21:03:37 -0600 |
---|---|---|
committer | Jack Sangdahl <[email protected]> | 2025-09-25 21:03:37 -0600 |
commit | 23353c144ac9b158af7616dc46b636d53b73ce34 (patch) | |
tree | 3394d416aa99e9dc1c84cc07e663e6cc3d844570 /valid_triangle_number.py | |
parent | afadd0bbf56f8aff8c5b397df08731c750302fd4 (diff) | |
download | leetcode-23353c144ac9b158af7616dc46b636d53b73ce34.tar.gz leetcode-23353c144ac9b158af7616dc46b636d53b73ce34.zip |
611. valid triangle number
Diffstat (limited to 'valid_triangle_number.py')
-rw-r--r-- | valid_triangle_number.py | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/valid_triangle_number.py b/valid_triangle_number.py new file mode 100644 index 0000000..8460292 --- /dev/null +++ b/valid_triangle_number.py @@ -0,0 +1,31 @@ +# 611. Valid Triangle Number + +""" +given an integer array 'nums', return the number of triplets chosen from the +array that can make triangles if we take them as side lengths of a triangle +""" + + +class Solution(object): + def triangleNumber(self, nums): + """ + :type nums: List[int] + :rtype: int + """ + nums.sort() + n, ans = len(nums), 0 + for k in range(2, n): + i, j = 0, k - 1 + while i < j: + if nums[i] + nums[j] > nums[k]: + ans += j - i + j -= 1 + else: + i += 1 + return ans + + +if __name__ == "__main__": + obj = Solution() + print(obj.triangleNumber(nums=[2, 2, 3, 4])) + print(obj.triangleNumber(nums=[4, 2, 3, 4])) |