大噶猴,前一段比较忙,刷题日记被耽搁了一段时间,从今天起开始恢复。
今天开始刷leetcode中等难度的算法题了,第一道是三数之和,看下题目要求:
思路:
好久没刷题了手非常生,思路也很枯竭,只想到了暴力循环,找到所有相加等于零的vector之后对每一个vector做一个排序然后判断是否相等,erase掉相同的元素。代码是这样的:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
vector<int> temp;
int len = nums.size();
int sum = 0;
for(int m = 0;m < len;m++){
for(int n = m + 1;n < len;n++){
for(int o = n + 1;o < len;o++){
sum = nums[m] + nums[n] + nums[o];
if(sum == 0){
temp.push_back(nums[m]);
temp.push_back(nums[n]);
temp.push_back(nums[o]);
result.push_back(temp);
temp.clear();
}
sum = 0;
}
}
}
int l = result.size();
for(int x = 0;x < l;x++){
sort(result[x].begin(),result[x].end());
}
for(int y = 0;y < l;y++){
for(int z = y + 1;z < l;z++){
if(result[y] == result[z]){
result.erase(result.begin() + z);
l--;
z--;
}
}
}
return result;
}
};
可以成功跑通,但是超出了时间限制,非常悲伤。
然后借鉴了别人的思路,可以对输入的数字先做一个排序,对负数部分进行循环,看其后面的两数相加是否能与它抵消,当然如果第一个数就是整数就是无解的输入。在计算后面两数的和的时候我们可以设两个迭代器,一个从这个数后第一个数开始,一个从最后一个数开始,如果加和小了就移动第一个迭代器,如果加和大了就移动第二个迭代器,就可以遍历所有情况了。而且因为数字整个都是排序的,如果有重复数字肯定是相邻的,我们在处理完前一个数之后,如果后一个数与之相等就删掉,这样避免了重复情况的出现。代码如下:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
sort(nums.begin(), nums.end());
for (int k = 0; k < nums.size(); ++k) {
if (nums[k] > 0) break;
if (k > 0 && nums[k] == nums[k - 1]) continue;
int target = 0 - nums[k];
int i = k + 1, j = nums.size() - 1;
while (i < j) {
if (nums[i] + nums[j] == target) {
res.push_back({nums[k], nums[i], nums[j]});
while (i < j && nums[i] == nums[i + 1]) ++i;
while (i < j && nums[j] == nums[j - 1]) --j;
++i; --j;
} else if (nums[i] + nums[j] < target) ++i;
else --j;
}
}
return res;
}
};
可见刷题是不能断的,感觉自己退步了,sad,还有很多要学习,加油!
我们下期见!