目标
给定一个包含非负整数的数组 nums ,返回其中可以组成三角形三条边的三元组个数。
示例 1:
输入: nums = [2,2,3,4]
输出: 3
解释:有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
示例 2:
输入: nums = [4,2,3,4]
输出: 4
说明:
- 1 <= nums.length <= 1000
- 0 <= nums[i] <= 1000
思路
有一个非负整数的数组,返回其中可以组成三角形的三元组。
可以先排序,然后使用二重循环,二分查找满足条件的第三边。
代码
/**
* @date 2025-09-26 8:52
*/
public class TriangleNumber611 {
public int triangleNumber(int[] nums) {
int n = nums.length;
Arrays.sort(nums);
int res = 0;
for (int i = 0; i < n; i++) {
for (int j = i + 1; j < n; j++) {
int t = nums[i] + nums[j];
int index = bs(nums, j, t);
res += Math.max(0, index - j);
}
}
return res;
}
public int bs(int[] nums, int l, int target) {
int r = nums.length - 1;
int m = l + (r - l) / 2;
while (l <= r) {
if (nums[m] >= target) {
r = m - 1;
} else {
l = m + 1;
}
m = l + (r - l) / 2;
}
return r;
}
}