题目
Given an array consists of non-negative integers, your task is to count the number of triplets chosen from the array that can make triangles if we take them as side lengths of a triangle.
- Example 1:
Input: [2,2,3,4]
Output: 3
Explanation:
Valid combinations are:
2,3,4 (using the first 2)
2,3,4 (using the second 2)
2,2,3
- Note:
The length of the given array won’t exceed 1000.
The integers in the given array are in the range of [0, 1000].
#分析题目
- 给了一个数组,问能组成多少个正确的三角形,根据三角形的性质,任意两条边之和要大于第三边。
- 问题就变成了找出所有这样的三个数字,使得任意两个数字之和都大于第三个数字。那么三个数字中如果较小的两个数字之和大于第三个数字,那么任意两个数字之和都大于第三个数字.
- 思路是排序之后,从数字末尾开始往前遍历,将left指向首数字,将right之前遍历到的数字的前面一个数字,然后如果left小于right就进行循环,循环里面判断如果left指向的数加上right指向的数大于当前的数字的话,那么right到left之间的数字都可以组成三角形,这相当于此时确定了i和right的位置,可以将left向右移到right的位置,中间经过的数都大于left指向的数,所以都能组成三角形.加完之后,right自减一,即向左移动一位。如果left和right指向的数字之和不大于nums[i],那么left自增1,即向右移动一位.优化时间复杂度到0(n2)(见解法二).
#解法一 Brute Force(Time Limit Exceeded)
- 时间复杂度O(N3 )
- 空间复杂度O(1)
#代码
public class Solution {
public int triangleNumber(int[] nums) {
int count = 0;
for (int i = 0; i < nums.length - 2; i++) {
for (int j = i + 1; j < nums.length - 1; j++) {
for (int k = j + 1; k < nums.length; k++) {
if (nums[i] + nums[j] > nums[k] && nums[i] + nums[k] > nums[j] && nums[j] + nums[k] > nums[i])
count++;
}
}
} I
return count;
}
}
#解法二 双指针
- 时间复杂度O(N2 )
- 空间复杂度O(log N) . Sorting takes O(log N) space.
public class Solution {
public int triangleNumber(int[] nums) {
int res = 0, n = nums.length;
int count = 0;
if(n == 0) return 0;
Array.sort(nums);
for(int k = n-1;k >= 0;k--){
int i = 0;
int j = k-1;
while(i < j){
if(nums[i] + nums[j] > nums[k]) {
count += j-i;
j--;
}
else i++;
}
}
return count;
}
}