58 四数之和

Posted 2020-11-10 唐的糖

原题网址：https://www.lintcode.com/problem/4sum/description

描述

给一个包含n个数的整数数组S，在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。

四元组(a, b, c, d)中，需要满足a <= b <= c <= d

答案中不可以包含重复的四元组。

您在真实的面试中是否遇到过这个题？  是

样例

例如，对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为：

(-1, 0, 0, 1)

(-2, -1, 1, 2)

(-2, 0, 0, 2)

 

标签

 

排序

哈希表

两根指针

数组

 

思路：沿用三数之和的方法，对数组排序，然后固定两个数，再设置两个指针左右滑动，时间复杂度为O（n^3）。注意每个索引都要去除重复。

 

AC代码：

class Solution {
public:
    /**
     * @param numbers: Give an array
     * @param target: An integer
     * @return: Find all unique quadruplets in the array which gives the sum of zero
     */
    vector<vector<int>> fourSum(vector<int> &numbers, int target) {
        // write your code here
        vector<vector<int>> result;
    int n=numbers.size();
    sort(numbers.begin(),numbers.end());
    for (int i=0;i<n;i++)
    {
        if (i>0&&numbers[i]==numbers[i-1])
        {
            continue;
        }
        for (int j=i+1;j<n;j++)
        {
            if (j>i+1&&numbers[j]==numbers[j-1])
            {
                continue;
            }
            int left=j+1,right=n-1;
            while(left<right)
            {
                if (left>j+1&&numbers[left]==numbers[left-1])
                {
                    left++;
                    continue;
                }
                if (right<n-1&&numbers[right]==numbers[right+1])
                {
                    right--;
                    continue;
                }
                int sum=numbers[i]+numbers[j]+numbers[left]+numbers[right];
                if (sum==target)
                {
                    vector<int> temp(4,numbers[i]);
                    temp[1]=numbers[j];
                    temp[2]=numbers[left];
                    temp[3]=numbers[right];
                    result.push_back(temp);
                    left++;
                    right--;
                }
                else if (sum<target)
                {
                    left++;
                }
                else
                {
                    right--;
                }
            }
        }
    }
    return result;
        
        
    }
};