LeetCode 611. Valid Triangle Number

和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

猜你喜欢

转载自www.cnblogs.com/hankunyan/p/11525332.html