LeetCode 611. Valid Triangle Number

Posted 2022-11-03 hankunyan

和3Sum那题非常类似。3Sum我们是固定i，令j=i+1, k=n-1。

本题由于三角形需要两边之和大于第三边，nums[i]+nums[j]>nums[k]。因此我们可以固定k，令i=0, j=k-1。

如果nums[i]+nums[j]>nums[k]，说明 [i,j-1] 范围内的所有数作为i，加上j和k都是可以的。

反之如果nums[i]+nums[j]≤nums[k]，那我们必须增加nums[i]，和才会增大。

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

时间复杂度 O(n^2)

 

Reference

https://leetcode.com/problems/valid-triangle-number/discuss/128135/A-similar-O(n2)-solution-to-3-Sum