[算法学习01]相向双指针的使用

强风吹拂y

已于 2024-03-21 21:58:33 修改

阅读量987

点赞数 32

分类专栏：算法学习文章标签：算法学习

于 2023-12-04 17:21:37 首次发布

本文链接：https://blog.csdn.net/weixin_45066884/article/details/134763546

版权

算法学习专栏收录该内容

8 篇文章 0 订阅

订阅专栏

前言

跟着灵茶山艾府大佬学算法！

两数之和三数之和【基础算法精讲 01】_哔哩哔哩_bilibili

使用相向双指针的前提：

数组已经排序完毕

提示：以下是本篇文章正文内容，下面案例可供参考

一、Leetcode167. 两数之和 II - 输入有序数组

给你一个下标从 1 开始的整数数组 numbers ，该数组已按 非递减顺序排列 ，请你从数组中找出满足相加之和等于目标数 target 的两个数。如果设这两个数分别是 numbers[index1] 和 numbers[index2] ，则 1 <= index1 < index2 <= numbers.length 。

以长度为 2 的整数数组 [index1, index2] 的形式返回这两个整数的下标 index1 和 index2。

你可以假设每个输入 只对应唯一的答案 ，而且你 不可以 重复使用相同的元素。

你所设计的解决方案必须只使用常量级的额外空间。

示例 1：

输入：numbers = [2,7,11,15], target = 9
输出：[1,2]
解释：2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。

示例 2：

输入：numbers = [2,3,4], target = 6
输出：[1,3]
解释：2 与 4 之和等于目标数 6 。因此 index1 = 1, index2 = 3 。返回 [1, 3] 。

示例 3：

输入：numbers = [-1,0], target = -1
输出：[1,2]
解释：-1 与 0 之和等于目标数 -1 。因此 index1 = 1, index2 = 2 。返回 [1, 2] 。

解题思路 使用相向双指针

left=0，right=len（numbers），s = numbers[left] + numbers[right]

因为数组本身已经按照从小到大排列好顺序了，因此如果s>target，则可以通过right = right - 1调整使s变小，如果s < target，则通过left = left + 1，使s变大。直到找到s == taget。

class Solution:

    def twoSum(self, numbers: List[int], target: int) -> List[int]:

        left, right = 0, len(numbers)-1

        while left < right:

            s = numbers[left] + numbers[right]

            if s == target:

                return left+1, right+1

            elif s > target:

                right = right - 1

            else:

                left = left + 1

二、Leetcode15. 三数之和

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

解题思路 方法仍然是使用相向双指针

要求：现在是三数之和，要求是三个数相加的和等于0，并且这三个数的下标不能相同且不能重复出现相同的三元组。

按住一个数不动，然后去枚举第二个数和最后一个数进行相向双指针。这样做就可以将包含第一个数的满足要求的三元组全部求出。

然后依次按住第二个数、第三个数....一直到倒数第三个数。因为三元组要求包含三个数，所以取到倒数第三个数就可以停下来了。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        i = 0
        n = len(nums)
        ans = []
        while i < n - 2:
            # 为了防止出现重复的三元组
            if i > 0 and nums[i] == nums[i-1]:
                i = i + 1
                continue
            # 优化1： 三个最小的数相加都＞0，没必要继续循环了，后续的数怎么加都＞0。
            if nums[i] + nums[i+1] + nums[i+2] > 0: 
                break
            # 优化2：如果当前的数加上最后两个数，仍然<0，那么在[i,n-1]这个范围随便加都<0，故可以跳出。
            if nums[i] + nums[n-1] + nums[n-2] < 0:
                i = i + 1
                continue 
            j = i + 1
            k = n - 1
            while j<k:
                s = nums[i] + nums[j] + nums[k]
                if s > 0:
                    k = k - 1
                elif s < 0:
                    j = j + 1
                else:
                    ans.append([nums[i], nums[j], nums[k]])
                    j = j + 1
                    k = k - 1
                    # 去重     
                    while j<k and nums[j] == nums[j-1]:
                        j = j + 1
                    # 去重
                    while k>j and nums[k] == nums[k+1]:
                        k = k - 1
            i = i + 1
        return ans

三、Leetcode16. 最接近的三数之和

给你一个长度为 n 的整数数组 nums 和一个目标值 target。请你从 nums 中选出三个整数，使它们的和与 target 最接近。

返回这三个数的和。

假定每组输入只存在恰好一个解。

示例 1：

输入：nums = [-1,2,1,-4], target = 1
输出：2
解释：与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。

示例 2：

输入：nums = [0,0,0], target = 1
输出：0

解题思路 这题与上一题基本是一样都是要按住一个数，然后枚举后两个数！

只是这题要求返回的是和，所以不需要去重，因为返回的是一个数！

最接近可以用距离来描述 abs(s-target) 这个数越小，说明越接近。

class Solution:
    def threeSumClosest(self, nums: List[int], target: int) -> int:
       nums.sort()
       n = len(nums)
       distance = float('inf')
       ans = 0
       for i in range(n):
           j = i + 1
           k = n - 1
           
           while j < k:
               s = nums[i] + nums[j] + nums[k]
               if abs(s - target) < distance:
                   ans = s
                   distance = abs(s - target)
               if s > target:
                   k = k - 1
               elif s < target:
                    j = j + 1
               else:
                    return s 
       return ans

四、Leetcode18. 四数之和

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按 任意顺序 返回答案。

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2：

输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

解题思路 其实和三数之和的做法基本是一样的，只是这次要按住的是两个数。然后对后两个数使用相向双指针。

有一些小细节需要注意

按住 i 的时候，j的起始位置应该是i+1。
判断j是否重复的时候，判断条件应该是 j > i+1，当 j 刚开始等于 i+1 的时候，如果直接执行 if nums[j] == nums[j-1]:，会导致在 j 第一次移动时，就检查与前一个元素是否相等，从而可能跳过一些潜在的解。通过添加 j > i+1 的条件，确保了在 j 移动一次之后才会开始检查是否相等，避免了这个问题。

class Solution:
    def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
        nums.sort()
        n = len(nums)
        ans = []
        for i in range(n-3):
            if i > 0 and nums[i] == nums[i-1]:
                continue 
            for j in range(i+1, n-2):
                if j > i+1 and nums[j] == nums[j-1]:
                    continue 
                k = j + 1
                v = n - 1
                while k < v:
                    s = nums[i] + nums[j] + nums[k] + nums[v]
                    if s > target:
                        v = v - 1
                    elif s < target:
                        k = k + 1
                    else:
                        ans.append([nums[i], nums[j], nums[k], nums[v]])
                        k = k + 1
                        v = v - 1
                        while k < v and nums[k] == nums[k-1]:
                            k = k + 1
                        while v > k and nums[v] == nums[v+1]:
                            v = v - 1
        return ans

五、Leetcode2824. 统计和小于目标的下标对数目

给你一个下标从 0 开始长度为 n 的整数数组 nums 和一个整数 target ，请你返回满足 0 <= i < j < n 且 nums[i] + nums[j] < target 的下标对 (i, j) 的数目。

示例 1：

输入：nums = [-1,1,2,3,1], target = 2
输出：3
解释：总共有 3 个下标对满足题目描述：
- (0, 1) ，0 < 1 且 nums[0] + nums[1] = 0 < target
- (0, 2) ，0 < 2 且 nums[0] + nums[2] = 1 < target 
- (0, 4) ，0 < 4 且 nums[0] + nums[4] = 0 < target
注意 (0, 3) 不计入答案因为 nums[0] + nums[3] 不是严格小于 target 。

示例 2：

输入：nums = [-6,2,5,-2,-7,-1,3], target = -2
输出：10
解释：总共有 10 个下标对满足题目描述：
- (0, 1) ，0 < 1 且 nums[0] + nums[1] = -4 < target
- (0, 3) ，0 < 3 且 nums[0] + nums[3] = -8 < target
- (0, 4) ，0 < 4 且 nums[0] + nums[4] = -13 < target
- (0, 5) ，0 < 5 且 nums[0] + nums[5] = -7 < target
- (0, 6) ，0 < 6 且 nums[0] + nums[6] = -3 < target
- (1, 4) ，1 < 4 且 nums[1] + nums[4] = -5 < target
- (3, 4) ，3 < 4 且 nums[3] + nums[4] = -9 < target
- (3, 5) ，3 < 5 且 nums[3] + nums[5] = -3 < target
- (4, 5) ，4 < 5 且 nums[4] + nums[5] = -8 < target
- (4, 6) ，4 < 6 且 nums[4] + nums[6] = -4 < target

解题思路

弄明白一件事就好了。我们要的是两个数的和s < target，那么要做的事就是 s>=target的时候更新一下right， s<taget的时候记录一下ans并且更新left。

s >= target：right = right - 1
s < target ：① ans += right - left ②left = left + 1

对①的解释：当s < target时，意味着nums[left] + nums[right] < target

那么 nums[left] 与区间[left+1, right]的任意一个数组成的和都 < target。区间[left+1, right]的长度就是nums[left] 与区间[left+1, right]的任意一个数组成的对数，所以 right - (left+1) + 1 = right - left。

class Solution:
    def countPairs(self, nums: List[int], target: int) -> int:
        nums.sort()
        ans = 0
        n = len(nums)
        left = 0
        right = n - 1
        while left < right:
            s = nums[left] + nums[right]
            if s >= target:
                right = right - 1
            else:
                ans += right - left 
                left = left + 1
        return ans

六、Leetcode611. 有效三角形的个数

给定一个包含非负整数的数组 nums ，返回其中可以组成三角形三条边的三元组个数。

示例 1:

输入: nums = [2,2,3,4]
输出: 3
解释:有效的组合是: 
2,3,4 (使用第一个 2)
2,3,4 (使用第二个 2)
2,2,3

示例 2:

输入: nums = [4,2,3,4]
输出: 4

解题思路 这个题好难理解，最后看了大佬的题解才弄明白。

想要构成一个三角形，要求是 任意两边之和 > 第三边。

思路一: 直接暴力破解，我只有这个思路0 0，但是有些样例过不去。

def triangleNumber(self, nums: List[int]) -> int:
        n = len(nums)
        ans = 0
        for i in range(n-2):
            for j in range(i+1, n-1):
                for k in range(j+1, n):
                    if nums[i] + nums[j] > nums[k] and nums[i] + nums[k] > nums[j] and nums[j] + nums[k] > nums[i]:
                        ans += 1
        return ans

思路二：

首先将这个数组排好序。

然后枚举第三条边。将问题任意两边之和 > 第三边 --->变为 a + b > c

原理是如果数组有序（从小到大），那么如果 a+b > c，肯定满足任意两边之和 > 第三边。

因为 c是在 a、b的后边，所以 c > b > a ----> a+c > b 以及 b+c > a。

小细节：

枚举第三条边，就要保证前面至少有两个数，所以从下标2这个位置开始遍历。
if 前两条边的和 < 第三条边，那么移动第一条边的指针，使其+1，原因如下。

nums[left] + nums[right] < c，那么 nums[left] + nums[left+1] < c, nums[left]与在[left+1, right]这个区间的数相加都 < c，所以 left = left+1。

如果 nums[left] + nums[right] > c ，那么 nums[right]与在[left， right-1]这个区间的数相加都 > c，故 ans += right - left。

 def triangleNumber(self, nums: List[int]) -> int:
        nums.sort()
        n = len(nums)
        ans = 0
        for i in range(2, n):
            c = nums[i]
            k = 0
            v = i - 1
            while k < v:
                s = nums[k] + nums[v]
                if s > c:
                    ans += v - k
                    v = v - 1
                else:
                    k = k + 1


        return ans