有效三角形的个数--二分和双指针

Posted 2021-09-01 C_YCBX Py_YYDS

文章目录

• • 题目
  • 法一：二分优化一层循环 O(n^2*logn)
  • 解法二：双指针优化两层循环 O(n^2)

题目

OJ平台

法一：二分优化一层循环 O(n^2*logn)

枚举两个较小值的和，二分寻找比这个值小的有边界。

class Solution {
public:
    int triangleNumber(vector<int>& nums) {
        int n = nums.size();
        if(n<3)return 0;
        int res = 0;
        sort(nums.begin(),nums.end());
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                int target = nums[i]+nums[j];
                int l = j+1,r = n;
                while(l<r){
                    int mid = (l+r)/2;
                    if(nums[mid]<target){
                        l = mid+1;
                    }
                    else
                        r = mid;
                }
                res += (l-j-1);
            }
        }
        return res;
    }
};

解法二：双指针优化两层循环 O(n^2)

最外层枚举最大边，里层双指针滑动。

class Solution {
public:
    int triangleNumber(vector<int>& nums) {
        int n = nums.size();
        if(n<3)return 0;
        int res = 0;
        sort(nums.begin(),nums.end());
        for (int i = n - 1; i >= 2; --i) {
            int l = 0, r = i - 1;
            while (l < r) {
                if (nums[l] + nums[r] > nums[i]) {
                    res += r - l;
                    --r;
                } else {
                    ++l;
                }
            }
        }
        return res;
    }
};