14_leetcode_fourSum

最新推荐文章于 2020-08-06 22:36:31 发布

原创最新推荐文章于 2020-08-06 22:36:31 发布 · 428 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #编程语言 #面试

leetcode 同时被 3 个专栏收录

155 篇文章

订阅专栏

算法

155 篇文章

订阅专栏

编程

155 篇文章

订阅专栏

本文介绍了一种解决四数之和问题的有效算法。该算法首先对输入数组进行排序，然后通过固定两个元素并寻找另外两个元素的方式找到所有可能的四元组，使得这四个数的和等于目标值。文章详细解释了如何避免重复解和如何优化搜索过程。

Given an array S of n integers, are there elements a, b, c, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.

Note:

Elements in a quadruplet (a,b,c,d) must be in non-descending order. (ie, a ≤ b ≤ c ≤ d)
The solution set must not contain duplicate quadruplets.

1.数组中元素个数小于4的时候；2:对数组元素进行排序；3从前到后遍历数组，注意数字相同的情况

    vector<vector<int> > fourSum(vector<int> &num, int target)
    {
        vector<vector<int> > result;
        if(num.size() <= 3)
            return result;
        
        sort(num.begin(), num.end());
        vector<int> quadruplet;
        int size = (int)num.size();
        
        for(int i = 0; i < size - 3; i++)
        {
            if(i == 0 || num[i] != num[i-1])
            {
                quadruplet.push_back(num[i]);
                fourSumCore(num, target - num[i], i + 1, quadruplet, result);
                quadruplet.pop_back();
            }
        }
        return result;
    }
    
    void fourSumCore(vector<int>& num, int target, int index, vector<int> &temp, vector<vector<int>> &result)
    {
        int size = (int)num.size();
        
        for(int i = index; i < size - 2; i++)
        {
            if(i == index || num[i] != num[i-1])
            {
                temp.push_back(num[i]);
                int leftValue = target - num[i];
                int left = i + 1;
                int right = size - 1;
                
                while(left < right)
                {
                    if(num[left] + num[right] == leftValue)
                    {
                        temp.push_back(num[left]);
                        temp.push_back(num[right]);
                        result.push_back(temp);
                        temp.pop_back();
                        temp.pop_back();
                        
                        right--;
                        while(left < right && num[right] == num[right + 1])
                            right--;
                    }
                    else if(num[left] + num[right] > leftValue)
                    {
                        right--;
                    }
                    else
                    {
                        left++;
                    }
                }
                temp.pop_back();
            }
        }
    }