关于Leetcode的刷题方法,在网上看了很多,每个人都有着不同的见解。作为Leetcode小白,我还是决定用最原始的方法进行刷题,即把每一题的思路都记录下来,虽然比较耗时间,但是个人感觉这样子的方法会让我对题目有更深入的了解。希望能在该专栏中记录下刷题过程中遇到的大大小小的问题🍊。
题目描述
解题思路
- 在”两数之和“的博客中,我们引入了通过哈希表的方式来降低时间复杂度。而这里的三数之和,实际上是两数之和的进阶版。
- 由题意得,我们需要找到所有相加为0的三元组,且要求不重复。
- TIPS:如果题目中看到不重复的字眼,应当考虑排序,因为乱序的一个数组很难判断是否重复,需要引入其他东西来判断重复元素,较复杂。
- 下面我们通过最容易想到的暴力法,来一步步分析,如何在暴力法的基础上引入双指针的思想进行时间复杂度的优化。
暴力法
- 首先将原数组进行排序。
- 使用一个三重循环,遍历所有情况,同时对于重复的元素不重复考虑
- 该方法的时间复杂度为O(n3)
暴力法优化
- 对于一个排序好的数组a,我们使用三个指针 i,j,k分别指向三个数(i<j<k),由于是升序,且我们要找的是三个指针指向的值相加为0,因此下面的两种情况只要出现可直接跳出当前层的循环:
- 如果 i 指针指向的值大于0,则说明j,k指向的值也大于0,则三数相加必然大于0,即必然无解,结束程序。
- 如果 i 指针指向的值与 j 指针指向的值相加大于0,则说明不管 k 指针指向的值是什么,都无解,跳出 j 所在的循环。
双指针法
- 其实所谓双指针,不过就是指暴力法中通过 j 和 k 决定的值。
- 回顾一下两数之和的题目,我们需要找的是两个数相加等于目标值的所有情况。而这一题中的目标值是会随着 i 指针指向的值所改变。
- 因为我们要找的是
nums[i] + nums[j] + nums[k] = 0
,而双指针是j和k,因此转换一下得nums[j] + nums[k] = -nums[i] = target
,这样子就转化成了之前两数之和的那种情形,可通过哈希表的方式进行操作。但这里哈希表并不是最快的方法。 - 对于每一个 i 的大循环,我们令j = i + 1,k = nums.size() - 1,即让这两个指针指向两个边界。
- 由于k指向数组的最右边界,即表示最大的值,假设此时
nums[j] + nums[k] < target
,则表示k所在的第三重循环必然无解,直接跳出第三重循环,以此将第三重循环的复杂度降为O(1)。 - 如果
nums[j] + nums[k] = target
,则存入当前结果即可。
题目代码
1.排序+双指针(耗时较短)
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result; //存放最终结果
sort(nums.begin(), nums.begin() + nums.size()); //对原数组进行升序排序
for (int i = 0; i < nums.size(); i++) { //第一重循环:i
if (nums[i] > 0) //如果三个数中的最小值大于0,直接退出
break;
if (i > 0 && nums[i] == nums[i - 1]) { //跳过重复访问的元素
continue;
}
int k = nums.size() - 1; //初始化当前情况下双指针的右指针位置
for (int j = i + 1; j < nums.size(); j++) { //第二重循环
if (nums[i] + nums[j] > 0) //如果三个数中的较小两个数加起来大于0,直接退出
break;
if (j > i + 1 && nums[j] == nums[j - 1]) //跳过重复访问的情况
continue;
for (; k > j; k--) { //开始双指针算法的核心:缩小范围
if (nums[i] + nums[j] + nums[k] < 0) { //由于k指向的都是尽可能大的值,因此若三数之和小于0,第三重循环直接退出
break;
}
else if (nums[i] + nums[j] + nums[k] == 0) {//若出现可行解,存入,并跳出第三重循环
result.push_back(vector<int>{nums[i], nums[j], nums[k]});
break;
}
}
}
}
return result;
}
};
2.排序+哈希表(耗时较长但能AC)
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result; //存放最终结果
sort(nums.begin(), nums.begin() + nums.size()); //对原数组进行排序
unordered_map<int, int> A; //定义哈希表
int target;
bool flag = 0;
for (int i = 0; i < nums.size(); i++) { //第一重循环
if (i > 0 && nums[i] == nums[i - 1]) { //跳过重复值
continue;
}
if (nums[i] > 0) //若三数之中的最小值大于0,直接退出
break;
A.clear(); //清空上一轮双指针策略用过的哈希表
for (int j = i + 1; j < nums.size(); j++) { //第二重循环
if (flag == 1) { //表示找到过一次结果,以防重复存入相同结果
if (nums[j] == nums[j - 1]) {
continue;
}
else {
flag = 0;
}
}
target = -(nums[i] + nums[j]); //与两数之和题目中的target一样
if (A[target] > 0) { //若存在解,则存入结果
result.push_back(vector<int>{nums[i], target, nums[j]});
flag = 1;
}
else { //更新哈希表
A[nums[j]]++;
}
}
flag = 0; //存放结果标志位归0
}
return result; //返回最终结果
}
};