题目描述
给定一个包含非负整数的数组,你的任务是统计其中可以组成三角形三条边的三元组个数。
示例 1:
输入: [2,2,3,4]
输出: 3
解释:
有效的组合是:
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3
注意:
数组长度不超过1000。
数组里整数的范围为 [0, 1000]。
需要确定三个数字,首先确定其中两个,然后再判断剩余的那个,首先确定 i , j ,其中i > j,然后要在小于 j 的部分确定 k,构成的是三角形,那么任意两边之和大于第三边,也就是1)nums[i]+nums[j] >nums[k] ,2)nums[i]+nums[k] >nums[j],3)nums[j] +nums[k] >nums[i] ,前两个条件显然成立,因为根据我们遍历的条件就可以知道了,只需要在遍历过程中确定第三条。也就是如果nums[k]>nums[i]-nums[j] 此时k值是满足条件的,反之不满足条件。注意while循环当满足循环中的条件继续循环,否则就跳出循环,跳出循环的k值就是满足 3)条件的值的起始位置,计算的结果是 ( j-1)-k +1 =j-k ;当然也可能 k==j 此时对结果贡献为0.
class Solution {
public:
int triangleNumber(vector<int>& nums) {
sort(nums.begin(),nums.end());
int res=0;
for(int i=0;i<nums.size();i++)
{
for(int j=i-1,k=0;j>0 && k<j ; j--)
{
while(k<j && nums[k]<=nums[i]-nums[j]) k++;
res+=max(j-k,0);
}
}
return res;
}
};