LeetCode算法系列---【双指针】

    科技2024-07-03  75

    文章目录

    双指针应用一、左右指针/同向指针1. 两数之和(有序数组)2. 归并两个有序数组3. 去除指定元素(in-place)4. 相似题目 二、快慢指针1. 判断链表中是否有环2. 已知链表中有环,寻找环的起始位置3. 寻找链表的中点 4. 相似题目


    双指针应用

    双指针主要应用于遍历数组,两个指针指向不同的元素,协同完成任务;常见的指针类型有左右指针与快慢指针。


    一、左右指针/同向指针

    左右指针的应用较为广泛,一般数组多为有序数组,其中二分查找(元素查找)与滑动窗口算法涉及的题目较多,后续将对这两个算法进行详细介绍。本小节重点介绍左右指针的其他应用,如两数之和、反转数组、去除指定元素等。 【通用题解】 ① 元素搜索 通常两个指针为反向,即初始化为0与len(nums)-1;此时待搜索数组一般为有序数组。 左右遍历数组,满足条件时返回结果;不满足条件,则依据目标任务更新左指针(left+1)或右指针(right-1)。 ② in-place修改数组 通常用于移动数组元素,为优化逐元素移动的复杂度,引入双指针,通过交换左右指针指向的元素,直接将满足/不满足要求的元素移至前端/末端。

    1. 两数之和(有序数组)

    题目描述: 解题思路: 设置左右指针: ① 若两数之和与目标值相等,则返回左右指针的索引(+1); ② 若两数之和小于目标值,则左指针+1;否则,右指针-1。

    代码实现:

    class Solution: def twoSum(self, nums: List[int], target: int) -> List[int]: left, right = 0, len(nums)-1 while left < right: sum_i = nums[left] + nums[right] if sum_i == target: return [left+1, right+1] elif sum_i < target: left += 1 else: right -= 1

    2. 归并两个有序数组

    题目描述: 解题思路: 设置同向双指针:从后往前对两数组进行比较 若从前往后进行比较,则前面每增加一个元素,其后边的元素相应地均需往后移动一位;因此优先满足末端元素的排序,同时也考虑到数组nums1后n位数字为0,直接覆盖后n位数字,不会影响nums1前m位本身的顺序。 ① 若指向nums2的索引t2小于0,则说明nums2已经全部遍历,直接返回结果; ② 若指向nums1的索引t1小于0;或者nums1[t1] < nums2[t2],则替换值为nums[t2],此时nums2[t2]已经遍历,因此索引t2减1;否则替换值为nums1[t1],索引t1减1。

    代码实现:

    class Solution: def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None: """ Do not return anything, modify nums1 in-place instead. """ t1, t2 = m-1, n-1 i = m+n-1 while i >= 0: if t2 < 0: return else: if nums1[t1] < nums2[t2] or t1 < 0: nums1[i] = nums2[t2] t2 -= 1 else: nums1[i] = nums1[t1] t1 -= 1 i -= 1

    3. 去除指定元素(in-place)

    题目描述: 解题思路: 简单直接的解法:for循环遍历,若在数组中,则添加至结果数组;否则,继续进行遍历。此时空间复杂度为O(n)。 优化空间复杂度为O(1):设置左右指针,若元素满足条件(即需要移除的元素),则将左右指针对应的元素进行交换,即可将val直接移至最末端。

    class Solution: def removeElement(self, nums: List[int], val: int) -> int: i, j = 0, len(nums)-1 while i < j: if nums[i] == val: nums[i] = nums[j-1] j -= 1 else: i += 1 # print(i, j, nums) return j

    4. 相似题目

    ① 3sum 4sum ② 去除指定元素II (参考题目3)


    二、快慢指针

    快慢指针最经典的应用便是寻找链表中的环了,可扩展为寻找链表环的起点,同时链表的应用还有寻找链表中点、倒数第K个元素等。除此之外,快慢指针可看作同向指针,可应用于归并两个数组。

    【通用题解】 链表:快慢指针一般均初始化指向链表的头结点head,前进时fast指针在前,慢指针slow在后;通过设计fast指针的步长(1、2或K)来找到目标结点的位置。 数组:同向指针初始化索引值均为0或len(nums)-1。

    1. 判断链表中是否有环

    题目描述: 解题思路: 单链表的特点是每个节点只知道下一个节点,因此一个指针的话无法判断链表中是否含有环的,此时则需双指针来进行判断。 ① 链表中不含环:快指针最终会遇到空指针null,则可说明该链表不含环。 ② 链表中含有环:若仅使用单指针就会陷入死循环,因为环形数组中没有 null 指针作为尾部节点。 经典解法就是用两个指针,快指针步长为2,慢指针步长为1,快指针会超慢指针,并和慢指针相遇,则说明链表含有环。

    代码实现:

    class Solution: def hasCycle(self, head: ListNode) -> bool: if not head or not head.next: return False slow = head fast = head.next while slow != fast: if not fast or not fast.next: return False slow = slow.next fast = fast.next.next return True

    2. 已知链表中有环,寻找环的起始位置

    题目描述:

    解题思路: 代码实现:

    class Solution: def detectCycle(self, head: ListNode) -> ListNode: if not head: return None fast = slow = head # 初始化快慢指针 while fast and fast.next: slow = slow.next fast = fast.next.next if slow == fast: fast = head while fast: if slow == fast: return slow slow = slow.next fast = fast.next return None

    3. 寻找链表的中点

    题目描述: 解题思路: 设计快慢指针,快指针步长为2,慢指针步长为1,则当快指针走完整个链表时,慢指针正好指向链表的中点。

    代码实现:

    class Solution: def middleNode(self, head: ListNode) -> ListNode: fast = slow = head while fast and fast.next: fast = fast.next.next slow = slow.next return slow

    4. 相似题目

    ① 寻找链表倒数第k个数(剑指offer-面试题22) ② 寻找重复数字 数组中仅有一个重复元素!则可看作题目2(寻找环的起点)的变形,重复数字便可看作环的起点。


    Processed: 0.009, SQL: 8