LeetCode 018 4sum

来源：互联网发布：fx2n编程手册编辑：程序博客网时间：2024/06/16 05:46

和3sum一样的意思，给一个序列和一个目标，在序列中找出不重复的四个数的和为目标的所有情况。

思路和3sum也一样，只是需要先将两个数相加，转化为3sum中的一个数，其他的就用3sum来做。

PS：开始是这样想的，但是优化没有这么变态，看了下discuss中的优化，简直了，leetcode中的优化感觉都是在很小很小的角落里优化的那种感觉，而且没有给定规定的时间复杂度，所以只能慢慢试了= =。。

class Solution {public:    vector<vector<int>> fourSum(vector<int>& nums, int target) {        vector<vector<int> >w;        vector<int> temp;        if(nums.size()<4)return w;        temp.resize(4);        sort(nums.begin(),nums.end());        for(int ii=0;ii<nums.size()-3;ii++){            if(nums[ii]+nums[ii+1]+nums[ii+2]+nums[ii+3]>target)break;//优化            if(ii&&(nums[ii]==nums[ii-1]))continue;            for(int jj=ii+1;jj<nums.size()-2;jj++){                if(nums[ii]+nums[jj]+nums[jj+1]+nums[jj+2]>target)break;//优化                if((jj>ii+1)&&nums[jj]==nums[jj-1])continue;                int r=nums.size()-1;                int l=jj+1;                int ans=target-(nums[ii]+nums[jj]);                while(l<r){                    if(nums[l]+nums[r]==ans){                       temp[0]=nums[ii];                       temp[1]=nums[jj];                       temp[2]=nums[l];                       temp[3]=nums[r];                       w.push_back(temp);                       l++;                       r--;                       while(l<nums.size()&&nums[l]==nums[l-1])l++;//检查重复的                       while(r>-1&&nums[r]==nums[r+1])r--;//检查重复的                    }                    while(nums[l]+nums[r]>ans)r--;                    while(nums[l]+nums[r]<ans)l++;                }            }        }        return w;    }};

0 0