58 四数之和

原题网址: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;
        
        
    }
};

 去除重复除了使用if——continue的方式,还可以用while循环:

while(left < right && numbers[left] == numbers[left+1])

{

left++;

}

while(left < right && numbe [right] == numbers[right-1])

{

righ--;

}

此外,还可以用set,参考:http://www.cnblogs.com/grandyang/p/4515925.html

C++中set的用法

其他方法:哈希表,通过增加空间复杂度来降低时间复杂度:https://www.cnblogs.com/eudiwffe/p/6282635.html

 

posted @ 2018-06-01 13:56  eeeeeeee鹅  阅读(174)  评论(0编辑  收藏  举报