- c++
解题思路
三数之和可以解构成在二数之和问题的基础上,将target值变成变化的值。对数组进行遍历,每一个元素的值就是target的取值。
过程中的小技巧:
为保证不重复,首先对数组进行排序,但不可去重。当target取值相同时,直接跳过即可,这是保证不重复需要注意的其中一点。另一点是在寻找两数之和满足target值时,若遇到相同元素(相同元素一定相邻),可直接跳过。
用map结构求满足条件的两数时,时间复杂度是O(NlgN);用hash_map时,时间负责度是O(N);用双指针的时间复杂度也是O(N).三数之和即在二数的基础上再乘N的复杂度。本人尝试过,若用map结构解该题会报超时。
用双指针是相对合适的解法,原因是双指针遍历时,可以结合先剪枝策略,从而降低平均耗时。
先剪枝策略:前面的指针所访问的元素大于target时,可提前结束。
过程中的注意事项:
边界判断:数组长度不得小于3
第一个元素大于0或最后一个元素小于0时,直接结束,返回空数组
作者:joyanmei-2
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res_final;
sort(nums.begin(), nums.end());
if (nums.size() < 3 || nums[0] > 0 || nums[nums.size()-1] < 0) return res_final;
for (size_t i = 0; i < nums.size(); ++i) {
if (nums[i] > 0) break;
if (i>0 && nums[i] == nums[i-1]) continue;
int target = 0-nums[i];
int k = i+1;
int m = nums.size()-1;
while(k<m) {
if (nums[k] > target) break;
if ((k>i+1 && nums[k] == nums[k-1]) || nums[k]+nums[m] < target) {
++k;
} else if ((m<nums.size()-1 && nums[m] == nums[m+1]) || nums[k]+nums[m] > target) {
--m;
} else {
vector<int> res;
res.push_back(nums[i]);
res.push_back(nums[k]);
res.push_back(nums[m]);
res_final.push_back(res);
++k;
}
}
}
return res_final;
}
};
- python
作者:lei_zm
# 原理
# 先将数组进行排序
# 从左侧开始逐渐选取一个值为定值
# 从右侧剩余的值中获取的与前面定值相加为 0 的两个值
# 首尾与定值相加
# 等于 00,记录这三个值
# 小于 00,首部右移
# 大于 00,尾部左移
# 定值右移,重复该步骤
class Solution:
def threeSum(self, nums):
# 从小到大排序
nums.sort()
li = []
length = len(nums)
# 从左侧开始逐渐选取一个值为定值
for i in range(length):
# 避免重复项,如 [-4, -1, -1, 0, 1, 2] 中由于两个 -1 导致的两个 [-1, 0, 1]
if i > 0 and nums[i] == nums[i-1]:
continue
# 减少无关计算,定值大于 0 后,后面的都大于 0,没必要进行计算了
if nums[i] > 0:
break
# 左指针随着定值的移动而移动
l = i + 1
r = length - 1
# 获取右侧剩余的值中的与前面定值相加为 0 的两个值
while l < r:
# 先求和,减少因 if 判断导致的重复计算和增加内存空间
s = nums[i] + nums[l] + nums[r]
# 满足条件,添加
if s == 0:
li.append([nums[i], nums[l], nums[r]])
# 避免重复项
while (l < r) and (nums[l+1] == nums[l]):
l += 1
while (l < r) and (nums[r-1] == nums[r]):
r -= 1
# 满足后现指针均需移动,避免重复和无效计算
l += 1
r -= 1
# 小于 0,左指针右移
elif s < 0:
l += 1
# 大于 0,右指针左移
else:
r -= 1
return li