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].
题意:给出数列nums,求随机取3个数能够组成的三角形的个数。
满足三角形:两条小边的和大于大边,因此首先进行排序。
重点:三重循环确定三边,最大边用二分法确定(当时面试时死活都没有想到的优化。。。。。。。)。
代码如下:
class Solution {
public int triangleNumber(int[] nums) {
Arrays.sort(nums);
int n=nums.length;
int count=0;
for(int i=0;i<n-2;i++)
{
for(int j=i+1;j<n-1;j++)
{
int l=j;
int r=n;
while(l+1<r){
int mid=(l+r)/2;
if(nums[i]+nums[j]<=nums[mid])
r=mid;
else
l=mid;
}
count+=l-j;
}
}
return count;
}
}