给定一个包含非负整数的数组,你的任务是统计其中可以组成三角形三条边的三元组个数。
示例 1:
输入: [2,2,3,4] 输出: 3 解释: 有效的组合是: 2,3,4 (使用第一个 2) 2,3,4 (使用第二个 2) 2,2,3
注意:
- 数组长度不超过1000。
- 数组里整数的范围为 [0, 1000]。
暴力解法 O(n^3) 小优化 break
class Solution {
public int triangleNumber(int[] nums) {
int n = nums.length;
if(n<3)
return 0;
Arrays.sort(nums);
int t=0;
for(int i=0;i<n-2;i++)
for(int j=i+1;j<n-1;j++)
for(int k=j+1;k<n;k++){
if(nums[i]+nums[j]>nums[k] && nums[k]-nums[j]<nums[i])
t++;
else
break;
}
return t;
}
}
beat 100% 双指针
class Solution {
public int triangleNumber(int[] nums) {
int n = nums.length;
int res = 0;
Arrays.sort(nums);
for (int i = n - 1; i >= 2; i--) {
int l = 0, r = i - 1;
while (l < r) {
if (nums[l] + nums[r] > nums[i]) {
res += r - l;
r--;
}else {
l++;
}
}
}
return res;
}
}