15.3Sum

来源:互联网 发布:淘宝客站内推广 编辑:程序博客网 时间:2024/06/05 10:55

给定一个包含n个整型元素的数组S,其中必定存在三个元素,它们的和为0,找出所有的S中所有满足和为0的唯一三元组
例子:

Input:  [-1, 0, 1, 2, -1, -4]Output:         [            [-1, 0, 1],            [-1, -1, 2]        ]

这道题也是3Sum类型,需要找出所有满足条件的三元组,3sum可以考虑:数组排序后,从第一个元素开始,确定第一个,然后在排在其后的有序数组上从两头遍历查找满足条件的数据。
解题思路:

  • 首先,先确定一个元素a,a依次从排序后的数组取值
    for(int i = 0;i < n -1; i++) a = S[i];
  • 其次,设置变量target = -a,在a后的有序数组中寻找和为target的两个元素。
  • 按照求有序数组两个数和为指定数的方法求解。

实现:

vector<vector<int>> threeSum(vector<int>& nums) {        sort(nums.begin(),nums.end());        vector<vector<int>> result;        for(int i = 0; i < nums.size();i ++)        {            if(i == 0 || nums[i - 1] != nums[i] && i > 0)            {                int target = -nums[i];                int m = i + 1;                int n = nums.size() - 1;                while(m < n)                {                    if(nums[m] + nums[n] > target)                    {                        n --;                    }                    else if(nums[m] + nums[n] < target)                    {                        m ++;                    }                    else                    {                        vector<int> tmp;                        tmp.push_back(nums[i]);                        tmp.push_back(nums[m]);                        tmp.push_back(nums[n]);                        result.push_back(tmp);                        while(nums[m] == nums[m + 1] && m < n)                            m++;                        while(nums[n] == nums[n - 1] && m < n)                            n--;                        m ++;                        n --;                    }                }            }        }        return result;    }
原创粉丝点击