leetcode18--四数之和

最新推荐文章于 2024-07-25 11:15:54 发布

_不会dp不改名_

最新推荐文章于 2024-07-25 11:15:54 发布

阅读量415

点赞数 7

分类专栏： # 双指针文章标签： leetcode 算法

本文链接：https://blog.csdn.net/bdn_nbd/article/details/136710329

版权

双指针专栏收录该内容

7 篇文章 0 订阅

订阅专栏

1. 题意

求所有数组中四个数之和为指定值的所有组合

四数之和

2. 题解

暴力没想到也能过

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        
        std::sort(nums.begin(), nums.end());
        vector<vector<int>> ans;

        int sz = nums.size();
        for (int i = 0;i < sz; ++i) 
        {
            if (i > 0 && nums[i] == nums[i - 1])
                    continue;
            for (int j = i + 1;j < sz; ++j) 
            {
                if (j > i + 1 && nums[j] == nums[j - 1])
                    continue;
                    int lp = j + 1;
                    int rp = sz - 1;

                    long long tot = (long long) nums[i] + nums[j];
                    while (lp < rp) 
                    {
                        long long rest = nums[lp] + nums[rp];
                        if ( tot + rest == target) {
                            ans.push_back({nums[i],nums[j],nums[lp],nums[rp]});
                            
                            ++lp;--rp;
                            while (lp< rp && nums[lp] == nums[lp-1]) {
                                ++lp;
                            }
                            while (lp < rp && nums[rp] == nums[rp+1]){
                                --rp;
                            }
                        }
                        else if (tot + rest  < target) {
                            lp++;
                        }
                        else {
                            rp--;
                        }
                    }

                
            }
        }


        return ans;
    }
};

双指针: 注意处理相等时的情况，需要左右指针都移到与上一位置不同的地方。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        
        std::sort(nums.begin(), nums.end());
        vector<vector<int>> ans;

        int sz = nums.size();
        for (int i = 0;i < sz; ++i) 
        {
            if (i > 0 && nums[i] == nums[i - 1])
                    continue;
            for (int j = i + 1;j < sz; ++j) 
            {
                if (j > i + 1 && nums[j] == nums[j - 1])
                    continue;
                    int lp = j + 1;
                    int rp = sz - 1;

                    long long tot = (long long) nums[i] + nums[j];
                    while (lp < rp) 
                    {
                        long long rest = nums[lp] + nums[rp];
                        if ( tot + rest == target) {
                            ans.push_back({nums[i],nums[j],nums[lp],nums[rp]});
                            
                            ++lp;--rp;
                            while (lp< rp && nums[lp] == nums[lp-1]) {
                                ++lp;
                            }
                            while (lp < rp && nums[rp] == nums[rp+1]){
                                --rp;
                            }
                        }
                        else if (tot + rest  < target) {
                            lp++;
                        }
                        else {
                            rp--;
                        }
                    }

                
            }
        }


        return ans;
    }
};