From 426b00a4e2889bb4c6d2599f29c965e173a5f536 Mon Sep 17 00:00:00 2001 From: Nada Date: Sun, 10 Apr 2022 14:56:56 +0800 Subject: [PATCH] =?UTF-8?q?Update=200018.=E5=9B=9B=E6=95=B0=E4=B9=8B?= =?UTF-8?q?=E5=92=8C.md?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 增加剪枝逻辑处理 --- problems/0018.四数之和.md | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/problems/0018.四数之和.md b/problems/0018.四数之和.md index 7304254e..ee70cb69 100644 --- a/problems/0018.四数之和.md +++ b/problems/0018.四数之和.md @@ -31,7 +31,7 @@ 四数之和，和[15.三数之和](https://programmercarl.com/0015.三数之和.html)是一个思路，都是使用双指针法, 基本解法就是在[15.三数之和](https://programmercarl.com/0015.三数之和.html) 的基础上再套一层for循环。 -但是有一些细节需要注意，例如： 不要判断`nums[k] > target` 就返回了，三数之和 可以通过 `nums[i] > 0` 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。（大家亲自写代码就能感受出来） +但是有一些细节需要注意，例如： 不要判断`nums[k] > target` 就返回了，三数之和 可以通过 `nums[i] > 0` 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。比如：数组是`[-4, -3, -2, -1]`，`target`是`-10`，不能因为`-4 > -10`而跳过。但是我们依旧可以去做剪枝，逻辑变成`nums[i] > target && (nums[i] >=0 || target >= 0)`就可以了。 [15.三数之和](https://programmercarl.com/0015.三数之和.html)的双指针解法是一层for循环num[i]为确定值，然后循环内有left和right下标作为双指针，找到nums[i] + nums[left] + nums[right] == 0。 @@ -72,15 +72,20 @@ public: vector> result; sort(nums.begin(), nums.end()); for (int k = 0; k < nums.size(); k++) { - // 这种剪枝是错误的，这道题目target 是任意值 - // if (nums[k] > target) { - // return result; - // } + // 剪枝处理 + if (nums[k] > target && (nums[k] >= 0 || target >= 0)) { + break; // 这里使用break，统一通过最后的return返回 + } // 去重 if (k > 0 && nums[k] == nums[k - 1]) { continue; } for (int i = k + 1; i < nums.size(); i++) { + // 2级剪枝处理 + if (nums[k] + nums[i] > target && (nums[k] + nums[i] >= 0 || target >= 0)) { + break; + } + // 正确去重方法 if (i > k + 1 && nums[i] == nums[i - 1]) { continue;