lintcode(leetcode)-----四数之和

题目描述:
给一个包含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)

思路讲解:首先读了题目之后,我的第一想法就是直接利用暴力法,嵌套循环四层,然后超时了,然后我就对其进行了一些剪枝,你如当你已经是第一个数字的4倍已经大于了目标值,这样下面就不用进行了,因为已经不可能的了(这里我们将数组进行排序了),剪枝过后还是发现超时,所以我们看了一下别人的思路,然后豁然开朗,他们是将查找最后两个数字的时候,采用的是双指针的方法,这样时间复杂度只有o(n)没有之前的两层循环o(n^2),然后这样就通过了,这里还需要注意的一点是题目要求不能有重复的,所以我们需要对其进行去重,所以我们要使用哈希表。最后就通过了。

代码详解:

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
        int length=numbers.size();
        sort(numbers.begin(),numbers.end());
        vector<vector<int>>res;
        vector<int>temp;
        set<vector<int> > s;//为了防止重复,用哈希表来避免

        for(int i=0;i<length;i++){
            for(int j=i+1;j<length;j++){
                int l=j+1;
                int r=length-1;
                int half_target=target-numbers[i]-numbers[j];

                while(l<r){//利用两根指针,来查找两个数的和是固定的某一个数
                    if(numbers[l]+numbers[r]==half_target){
                        temp.push_back(numbers[i]);
                        temp.push_back(numbers[j]);
                        temp.push_back(numbers[l]);
                        temp.push_back(numbers[r]);
                        s.insert(temp);
                        temp.clear();
                        l++;
                        r++;
                    }else if(numbers[l]+numbers[r]>half_target){
                        r--;
                    }else{
                        l++;
                    }
                }
            }
        }

        set<vector<int> >::iterator it;
        for(it=s.begin();it!=s.end();it++){
            res.push_back(*it);
        }

        return res;
    }

};

猜你喜欢

转载自blog.csdn.net/qq_34355232/article/details/79128952