https://leetcode-cn.com/problems/valid-triangle-number/
给定一个包含非负整数的数组,你的任务是统计其中可以组成三角形三条边的三元组个数。
示例 1:
输入: [2,2,3,4]
输出: 3
解释:
有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
注意:
数组长度不超过1000。
数组里整数的范围为 [0, 1000]。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/valid-triangle-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
我提交的code
/*
* @lc app=leetcode.cn id=611 lang=cpp
*
* [611] 有效三角形的个数
*/
// @lc code=start
class Solution {
public:
int triangleNumber(vector<int>& nums) {
int count = 0;
sort(nums.begin(), nums.end());
int lens = nums.size();
int tail = 0;
for (int i = 0; i < lens; i++) {
for (int j = i + 1; j < lens; j++) {
int sum = nums[i] + nums[j];
tail = lens - 1;
while(tail > j) {
if (nums[tail] >= sum)
tail--;
else
break;
}
int val = tail - j;
std::cout << "i:" << i << endl;
std::cout << "j:" << j << endl;
std::cout << "tail=" << tail << endl;
std::cout << "val=" << val << endl;
count += val;
}
}
return count;
}
};
// @lc code=end
test结果
Time Limit Exceeded
-
158/241 cases passed (N/A)
果然超时了,三遍for循环,时间复杂度。
官方给出的方案:
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) {
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 += k - j;
}
}
return ans;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/valid-triangle-number/solution/you-xiao-san-jiao-xing-de-ge-shu-by-leet-t2td/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
最里面的for循环变成了二分查找