LeetCode探索之旅(97)-18_4sum

今天继续刷LeetCode,第18题,求4个数的和。

分析:
类似于三个数,区别是多添加一个循环。然后通过双指针法,找到满足条件的组合。

问题:
1、注意要组合不一样,这样需要判断前后是否一样、

附上C++代码:

class Solution {
public:
    vector> fourSum(vector& nums, int target) {
        vector> re;
        sort(nums.begin(),nums.end());
        int len=nums.size();
        for(int i=0;i0&&nums[i]==nums[i-1])
                continue;
            for(int j=i+1;ji+1&&nums[j]==nums[j-1])
                    continue;
                int m=j+1;
                int n=len-1;
                while(mtarget)
                        n--;
                    else
                        m++;
                }
            }
        }
        return re;
    }
};

附上Python代码:

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        result = list()
        nums_len = len(nums)
        nums.sort()
        for i in range(nums_len - 3):
            if i > 0 and nums[i - 1] == nums[i]:
                continue

            for j in range(i+1, nums_len - 2):
                if j > i + 1 and nums[j] == nums[j - 1]:
                    continue
                l, r = j + 1, nums_len - 1
                while l < r:
                    if nums[i] + nums[j] + nums[l] + nums[r] == target:
                        result.append([nums[i], nums[j], nums[l], nums[r]])
                        while l < r and nums[l] == nums[l + 1]:
                            l += 1
                        while l < r and nums[r] == nums[r - 1]:
                            r -= 1
                        l += 1
                        r -= 1
                    elif nums[i] + nums[j] + nums[l] + nums[r] < target:
                        l += 1
                    else:
                        r -= 1 
        return result

你可能感兴趣的:(安装学习)