Triangle Numbers
DESCRIPTION (credit Leetcode.com)
Write a function to count the number of triplets in an integer array nums that could form the sides of a triangle. The triplets do not need to be unique.
EXAMPLES
Input:
nums = [11,4,9,6,15,18]
Output:
10
Explanation: Valid combinations are...
4, 15, 18
6, 15, 18
9, 15, 18
11, 15, 18
9, 11, 18
6, 11, 15
9, 11, 15
4, 6, 9
Run your code to see results here
Solution
def triangleNumber(nums):nums.sort()count = 0for i in range(len(nums) - 1, 1, -1):left = 0right = i - 1while left < right:if nums[left] + nums[right] > nums[i]:count += right - leftright -= 1else:left += 1return count
valid triangle numbers
0 / 26
1x
Explanation
In order for a triplet to be valid lengths of a triangle, the sum of any two sides must be greater than the third side. By sorting the array, we can leverage the two-pointer technique to count all valid triplets in O(n2) time and O(1) space.
The key to this question is realizing that if we have 3 numbers, such as 4, 8, 9, arranged from smallest to largest, and the sum of the two smallest numbers is greater than the largest number, then we have a valid triplet ( 4 + 8 > 9).
But not only that, triplets where the smallest number is between 4 and 8 are also valid triplets.
This means that if we sort the input array, and then iterate from the end of the array to the beginning, we can use the two-pointer technique to efficiently count all valid triplets.
The pointers i, left, and right represent the current triplet we are considering. If nums[i] + nums[left] >= nums[right], then we know there are a total of right - left valid triplets, since all triplets between left and right are also valid triplets. We can then decrement right to check for the valid triplets that can be made by decreasing the middle value.
def triangleNumber(nums):nums.sort()count = 0for i in range(len(nums) - 1, 1, -1):left = 0right = i - 1while left < right:if nums[left] + nums[right] > nums[i]:count += right - leftright -= 1else:left += 1return count
valid triangle numbers
0 / 5
1x
When nums[i] + nums[left] < nums[right], we know that all triplets between left and right are also invalid, so we increment left to look for a larger smallest value.
def triangleNumber(nums):nums.sort()count = 0for i in range(len(nums) - 1, 1, -1):left = 0right = i - 1while left < right:if nums[left] + nums[right] > nums[i]:count += right - leftright -= 1else:left += 1return count
Count:
4
move right pointer
0 / 1
1x
Each time left and right cross, we decrement i and reset left and right to their positions at opposite ends of the array. This happens until i is less than 2, at which point we have counted all valid triplets.
def triangleNumber(nums):nums.sort()count = 0for i in range(len(nums) - 1, 1, -1):left = 0right = i - 1while left < right:if nums[left] + nums[right] > nums[i]:count += right - leftright -= 1else:left += 1return count
Count:
4
move left pointer
0 / 20
1x
© 2024 Optick Labs Inc. All rights reserved.
Loading comments...