有效三角形的个数
题目链接
题目描述
:
给定一个包含非负整数的数组 nums ,返回其中可以组成三角形三条边的三元组个数
输入:
nums = [2,2,3,4]
输出:
3
思路1
:
排序+二分
:先排序,升序保证组成三角形的前两个条件a+c>b b+c>a- 二重循环(满足a<b)
- 二分从b的下一个开始,找到最大的满足条件即a+b>c 的c,这个位置到b位置的下一个(k-j)就是满足条件的三边情况
- 继续循环
思路2
:
双指针+二分 (逆向扫描)
:先排序,再逆序遍历nums数组(先确定最大边a,只要剩余的两边满足 b + c > a,则一定满足 a + c > b 和 a + b > c)- nums[i]为确定的a边,l=0(b边,左指针)r=i-1(c边,右指针)
- 结合二分while(l<r)
只要满足b + c > a就 r–, c边固定为nums[r],同时b边可以从l取到r-1共r-l种情况,cnt+=l-r;
否则l++注意:这里为什么不能用正向扫描?
答:正向扫描的时候,如果两边之和小于第三边,那么会有两种情况,一种是左指针往右移动,一种是右指针往左移动。反向扫描只能说左指针往右
代码如下 排序+二分:
class Solution { public: int triangleNumber(vector<int>& nums) { //排序+二分 sort(nums.begin(),nums.end());//less 升序保证组成三角形的前两个条件a+c>b b+c>a //剩下只需要满足a+b>c int cnt=0;//计数 for(int i=0;i<nums.size();i++)//二重循环(满足a<b) { int a=nums[i]; for(int j=i+1;j<nums.size();j++) { int b=nums[j]; int l=j+1;//从j+1开始 int r=nums.size()-1;//到数组末尾 int k=j;//最后满足c的数可能为0个 while(l<=r)//二分模板 { int mid=(l+r)/2; if(a+b>nums[mid]) { k=mid; l=mid+1; } else r=mid-1; } cnt+=(k-j);//最大满足条件的值j+1~max都是可以c取的值 } } return cnt; } };
代码如下 双指针+二分:
class Solution { public: int triangleNumber(vector<int>& nums) { sort(nums.begin(),nums.end()); int n=nums.size(); int cnt=0; for(int i=n-1;i>=2;i--) { int l=0; int r=i-1; while(l<r) { if(nums[l]+nums[r]>nums[i]) { cnt+=r-l; r--; } else l++; } } return cnt; } };