三角形计数
题目:
描述
给定一个整数数组,在该数组中,寻找三个数,分别代表三角形三条边的长度,问,可以寻找到多少组这样的三个数来组成三角形?
样例
样例 1:
输入: [3, 4, 6, 7]
输出: 3
解释:
可以组成的是 (3, 4, 6),
(3, 6, 7),
(4, 6, 7)
样例 2:
输入: [4, 4, 4, 4]
输出: 4
解释:
任何三个数都可以构成三角形
所以答案为 C(3, 4) = 4
解题思路:先将数组排序,然后遍历数组,假设当前遍历的是三角形中的最长边,再用双指针来寻找和大于该边的组合即可
public class Solution {
/**
* @param S: A list of integers
* @return: An integer
*/
public int triangleCount(int[] nums) {
// write your code here
int len = nums.length;
if(len < 3) {
return 0;
}
Arrays.sort(nums);
int ans = 0;
for(int i = 2; i < len; i++) {
int l = 0, r = i - 1;
while(l < r) {
int sum = nums[l] + nums[r];
if(sum > nums[i]) {
// nums[;] + nums[r] > nums[i]
// 说明nums[i + 1.. r - 1] + nums[r] > nums[i]
// 所以 ans += r - l 减少重复计算
ans += r - l;
r--;
} else {
l++;
}
}
}
return ans;
}
}