【leetcode】18. 四数之和（4sum）（模拟）[中等]

链接

https://leetcode-cn.com/problems/4sum/

耗时

解题：1 h 16 min
题解：6 min

题意

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

注意：

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

思路

把四数之和看成 两数之和+两数之和。先处理出所有的两数之和，包括值和位置，存入哈希表；然后枚举两数之和，在哈希表里找 target-两数之和，如果能找到，则加入结果。

对于去重问题，我用的是 四元组排序后放入 set 去重。😦 想不出更好的方法了。

时间复杂度：$\gg O(n^2)$

AC代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        int n = nums.size();
        unordered_map<int, vector<int>> unmp;
        for(int i = 0; i < n; ++i) {
            for(int j = i+1; j < n; ++j) {
                unmp[nums[i]+nums[j]].push_back(i);
                unmp[nums[i]+nums[j]].push_back(j);
            }
        }
        vector<vector<int>> ans;
        set<vector<int>> res;
        for(int i = 0; i < n; ++i) {
            for(int j = i+1; j < n; ++j) {
                int rest = target-nums[i]-nums[j];
                if(unmp.find(rest) != unmp.end()) {
                    auto tmp = unmp[rest];
                    for(int k = 0; k < tmp.size(); k+=2) {
                        if(tmp[k] == i || tmp[k] == j || tmp[k+1] == i || tmp[k+1] == j) 
                            continue;
                        vector<int> t = {nums[i],nums[j],nums[tmp[k]],nums[tmp[k+1]]};
                        sort(t.begin(), t.end());
                        res.insert(t);
                    }
                }
            }
        }
        for(auto r : res) {
            ans.push_back(r);
        }
        return ans;
    }
};