C++ 中的有效三角形数
假如我们有一个非负整数组成的数组,我们的任务是计算从数组中选出的那些三元组的数目,如果我们把它们作为三角形的边长,就能够组成三角形。因此,如果输入类似 [2,2,3,4],那么结果将是 3,因为 [2,3,4] 使用第一个 2,[2,3,4] 使用第二个 2,[2,2,3] 是三个三角形。
为了解决这个问题,我们将遵循以下步骤——
- ret := 0,n := num 的大小,nums 排序
- 对于范围 n – 1 到 0 的 i
- right := i – 1,left := 0
- while left < right
- sum := nums[left] + nums[right]
- 如果 sum > nums[i],则令 ret 增加 right – left 减小 right 1,否则增加 left 1
- 返回 ret
让我们看以下实现以获得更好的理解——
示例
#include <bits/stdc++.h> using namespace std; class Solution { public: int triangleNumber(vector<int>& nums) { int ret = 0; int n = nums.size(); sort(nums.begin(), nums.end()); for(int i = n - 1; i >= 0; i--){ int right = i - 1; int left = 0; while(left < right){ int sum = nums[left] + nums[right]; if(sum > nums[i]){ ret += right - left; right--; }else left++; } } return ret; } }; main(){ vector<int> v = {2,2,3,4}; Solution ob; cout << (ob.triangleNumber(v)); }
输入
[2,2,3,4]
输出
3
广告