diff --git a/problems/0015.三数之和.md b/problems/0015.三数之和.md index edd22ec1..0d47bf48 100644 --- a/problems/0015.三数之和.md +++ b/problems/0015.三数之和.md @@ -55,20 +55,20 @@ public: // 实际解法:a+b+c(存储)==0(检索) <=> b(存储)==0-(a+c)(检索) vector> threeSum(vector& nums) { // 本解法的内层循环一边存储一边检索,所以被存储的应该是b,而不是c - vector> res; + vector> result; sort(nums.begin(), nums.end()); - - // 如果只有正数,不可能形成和为0的三元组 - if (nums[0] > 0) - return res; for (int i = 0; i < nums.size(); i++) { + // 如果a是正数,a 0) + break; + // [a, a, ...] 如果本轮a和上轮a相同,那么找到的b,c也是相同的,所以去重a if (i > 0 && nums[i] == nums[i - 1]) continue; - // 这个st的作用是存储b - unordered_set st; + // 这个set的作用是存储b + unordered_set set; for (int k = i + 1; k < nums.size(); k++) { // [(-2x), ..., (x), (x), x, x, x, ...] @@ -81,18 +81,18 @@ public: // a+b+c=0 <=> b=0-(a+c) int target = 0 - (nums[i] + nums[k]); - if (st.find(target) != st.end()) { - res.push_back({nums[i], target, nums[k]}); // nums[k]成为c + if (set.find(target) != set.end()) { + result.push_back({nums[i], target, nums[k]}); // nums[k]成为c // 内层循环中,a固定,如果find到了和上轮一样的b,那么c也就和上轮一样,所以去重b - st.erase(target); + set.erase(target); } else { - st.insert(nums[k]); // nums[k]成为b + set.insert(nums[k]); // nums[k]成为b } } } - return res; + return result; } }; ```