【每日一题】611. 有效三角形的个数

611. 有效三角形的个数

难度中等271

给定一个包含非负整数的数组，你的任务是统计其中可以组成三角形三条边的三元组个数。

示例 1:

输入: [2,2,3,4]
输出: 3
解释:
有效的组合是: 
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3

注意:

1. 数组长度不超过1000。
2. 数组里整数的范围为 [0, 1000]。

解：

排序+二分查找 

 

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;
    }
};