欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

#leetcode刷题之路18-四数之和

程序员文章站 2023-09-09 08:23:38
给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。注意:答案中不可以包含重复的四元组。 示例:给定数组 nums = [1, ......

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

示例:
给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。
满足要求的四元组集合为:
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]

思路:暴力解法,暂时只对最后一个数做优化

#include <iostream>
#include<vector>
#include <algorithm>
#include <set>
using namespace std;
vector<vector<int>> foursum(vector<int>& nums, int target) {
    vector<vector<int>> ans;
    set<vector<int>> t;
    sort(nums.begin(),nums.end());
    int len=nums.size();
    for(int i=0;i<len-3;i++)
    {
        for(int j=i+1;j<len-2;j++)
        {
            for(int k=j+1;k<len-1;k++)
            {
                for(int h=k+1;h<len;h++)
                {
                    while ((h+1)<len&&nums[h]==nums[h+1]) h++;
                    if((nums[i]+nums[j]+nums[k]+nums[h])==target)
                    {
                        vector<int> temp;
                        temp.push_back(nums[i]);
                        temp.push_back(nums[j]);
                        temp.push_back(nums[k]);
                        temp.push_back(nums[h]);
                        int tt=t.size();
                        t.insert(temp);
                        if(tt!=t.size()) ans.push_back(temp);
                    }
                }
            }
        }
    }
    return ans;
}

int main() {
    vector<int> a={1,0,-1,0,-2,2};
    int target=0;
    vector<vector<int>> ans=foursum(a,target);
    cout<<ans.size()<<endl;
    return 0;
}