LintCode 58 四个数之和

来源:互联网 发布:而立不惑知天命的含义 编辑:程序博客网 时间:2024/05/17 09:23

题目:fourSum


要求:

给一个包含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)

算法要求:

解题思路:

跟上一题一样,直接暴力求解,但是四重循环后超时,所以改良一下,用前后查找法找最后二个数。

算法如下:

    vector<vector<int> > fourSum(vector<int> nums, int target)  {        // write your code here        vector<vector<int> > vec;        set<vector<int> > s;        sort(nums.begin(), nums.end());        int size = nums.size();        int mun;        for (int i = 0; i < size; i++) {            for (int j = i + 1; j < size; j++) {                mun = nums[i] + nums[j];                int k = j + 1, l = size - 1;                while (k < l) {                    int sum = mun + nums[k] + nums[l];                    if (sum == target) {                        vector<int> tempVec;                        tempVec.push_back(nums[i]);                        tempVec.push_back(nums[j]);                        tempVec.push_back(nums[k]);                        tempVec.push_back(nums[l]);                        s.insert(tempVec);                        k++;                        l--;                    } else if (sum < target) {                        k++;                    } else {                        l--;                    }                }            }        }        set<vector<int> >::iterator it;        for (it = s.begin(); it != s.end(); it++) {            vec.push_back(*it);        }        return vec;    }
0 0