题目
给定一个包含非负整数的数组,你的任务是统计其中可以组成三角形三条边的三元组个数。
示例 1:
输入: [2,2,3,4]
输出: 3
解释:
有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
注意:
数组长度不超过1000。
数组里整数的范围为 [0, 1000]。
题解1(C++)
排序 + 二分查找
class Solution {
public:
int triangleNumber(vector<int>& nums) {
int n = nums.size(); // 容器中变量的个数
sort(nums.begin(), nums.end()); // 对容器中的数字进行排序
int ans = 0; // 存放最终结果
// 二分法查找满足条件的最大的nums[mid] , [j + 1, k] 范围内的下标都可以作为边 c 的下标,将该范围的长度 k - j 累加入答案ans。
for(int i = 0; i < n; ++i){
for(int j = i +1 ; j< n ; ++j){
int left = j + 1, right = n - 1, k = j;
while(left <= right){
int mid = (left + right) / 2;
if(nums[mid] < nums[i] + nums[j]){
k = mid;
left = mid + 1;
}
else {
right = mid - 1;
}
}
ans = ans + k - j;
}
}
return ans;
}
};
题解2(C++)
排序 + 双指针(对题解一的优化)
class Solution {
public:
int triangleNumber(vector<int>& nums) {
int n = nums.size();
sort(nums.begin(), nums.end()); // 排序
int ans = 0;
for (int i = 0; i < n; ++i) {
int k = i;
// 双指针移动,j,k,得到满足条件的 max(k-j,0)进行累加
for (int j = i + 1; j < n; ++j) {
while (k + 1 < n && nums[k + 1] < nums[i] + nums[j]) {
++k;
}
ans += max(k - j, 0);
}
}
return ans;
}
};
注:所有题目以及代码均来自力扣,我只是对代码进行了注释,有时候会有一些自己的改动,加了一些自己的理解,仅供自己学习使用。如果大家对我的注释有什么疑问或者指正,欢迎大家一起讨论。