代码随想录算法训练营第四天 | 链表 part 2

enzoherewj

已于 2023-08-13 09:51:50 修改

阅读量147

点赞数

分类专栏：代码随想录算法训练营文章标签：算法链表数据结构 leetcode python

于 2023-08-12 12:11:19 首次发布

本文链接：https://blog.csdn.net/enzoherewj/article/details/132242320

版权

代码随想录算法训练营专栏收录该内容

52 篇文章 3 订阅

订阅专栏

24. 两两交换链表中的节点

Leetcode

给定一个链表，两两交换其中相邻的节点，并返回交换后的链表。
你不能只是单纯的改变节点内部的值，而是需要实际的进行节点交换。

如果链表是奇数个的话，那最后剩下的一个节点不需要处理。
链表为空则返回空

思路

使用虚拟头节点，这样处理每个节点的逻辑都是一样的。

我照着卡哥的思路，把步骤逐一画了出来，写的时候就清晰多了。

初始时，cur指向虚拟头结点，然后进行如下三步（橙色的三步），

待我们完成指针的转换之后，再将cur移动到temp1，方便进行下一次的链表节点两两互换。

代码

有一点值得注意的是，我一开始在写进入while loop条件的时候，并没有加cur.next，我直觉觉得只要cur.next.next能跑通就可以了。但是这样在遇到head为空的节点的时候就会出问题，因为cur.next is None，所以cur.next.next就会报错。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def swapPairs(self, head: Optional[ListNode]) -> Optional[ListNode]:
    	#虚拟头节点
        dummy = ListNode(next = head)

        cur = dummy

        while cur.next and cur.next.next:
            temp1 = cur.next # 防止节点修改
            temp3 = cur.next.next.next

            cur.next = cur.next.next
            cur.next.next = temp1
            temp1.next = temp3

            cur = temp1

        return dummy.next

复杂度分析

时间复杂度：O(n)
空间复杂度：O(1)

19.删除链表的倒数第N个节点

Leetcode

给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。

思路

双指针的经典应用，如果要删除倒数第n个节点，让fast移动n步，然后让fast和slow同时移动，直到fast指向链表末尾。删掉slow所指向的节点就可以了。

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeNthFromEnd(self, head: Optional[ListNode], n: int) -> Optional[ListNode]:
        dummy = ListNode(next = head)
        fast, slow = dummy.next, dummy
		
		#让fast指针先跑n次，确定与slow间隔的距离
        for i in range(n):
            fast = fast.next
            
		#slow, fast指针一起跑，fast跑到None的时候，slow刚好知道要移除的节点的前一个节点
        while fast:
            slow = slow.next
            fast = fast.next
        
        #删除指定节点
        slow.next = slow.next.next

        return dummy.next

复杂度分析

时间复杂度：O(n)
空间复杂度：O(1)

160.链表相交

Leetcode

给你两个单链表的头节点 headA 和 headB ，请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点，返回 null。

图示两个链表在节点 c1 开始相交：

思路

写法一

这道题可以像卡哥那样，先判断两条链表的长度，求差值，再末端对齐，开始curA和curB两个指针同时遍历到尾部。此时我们就可以比较curA和curB是否相同，如果不相同，同时向后移动curA和curB，如果遇到curA == curB，则找到交点。

否则循环退出返回空指针。

写法1

写法二

另外一个写法比较简洁，我在leetcode评论区看到一个完整的思路解析

Krahets-双指针，链表相交

具体的思路是初始化A, B = headA, headB。A指针遍历完headA链表之后再遍历headB链表，B指针遍历完headB链表之后再遍历headA链表。两个指针遍历的总长度是相同的，因为

len(A链表) + len(B链表) = len(B链表) + len(A链表)

在遍历的过程中判断A是否等于B，相等就返回A，即为相交点，否则当遍历完两条链表之后，A和B也会同时遍历到None的位置，这时候返回A，即为None，不存在相交点。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> Optional[ListNode]:

        A, B = headA, headB
        while A != B:
            A = A.next if A else headB
            B = B.next if B else headA

        return A

复杂度分析

时间复杂度: O(a+b)，两种写法在最差情况下，都需遍历 $a + b$ 个节点。
空间复杂度：O(1)，节点指针 A , B 使用常数大小的额外空间。

142.环形链表II

Leetcode

给定一个链表，返回链表开始入环的第一个节点。如果链表无环，则返回 null。

为了表示给定链表中的环，使用整数 pos 来表示链表尾连接到链表中的位置（索引从 0 开始）。如果 pos 是 -1，则在该链表中没有环。

说明：不允许修改给定的链表。

思路

在这里插入图片描述

$x = (n - 1) (y + z) + z$ ，这个公式意味着从头结点（A）出发一个指针，从相遇节点（B）也出发一个指针，这两个指针每次只走一个节点，那么当这两个指针相遇的时候就是环形入口的节点（C）。

代码

写法一

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def detectCycle(self, head: Optional[ListNode]) -> Optional[ListNode]:
        # method 1 fast and slow pointer, tortoise and hare algorithm 
        slow, fast = head, head

        while fast and fast.next:
            slow = slow.next
            fast = fast.next.next
            if slow == fast:
                slow = head

                while slow != fast:
                    slow = slow.next
                    fast = fast.next
            
                return slow
            
        return None

复杂度分析

时间复杂度：O(n)，快慢指针相遇前，指针走的次数小于链表长度，快慢指针相遇后，两个index指针走的次数也小于链表长度，总体为走的次数小于 2n
空间复杂度：O(1)

写法二

用set记录遍历过的节点，如果有重复的就返回。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def detectCycle(self, head: Optional[ListNode]) -> Optional[ListNode]:
        #method 2
        visited = set()

        while head:
            if head in visited:
                return head
            visited.add(head)
            head = head.next

        return None

复杂度分析

时间复杂度：O(n)
空间复杂度：O(n)

21. 合并两个有序链表（复习下）

Leetcode

将两个升序链表合并为一个新的升序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

思路

使用虚拟头节点
- 初始化简单，省去了要考虑是取红链表为头节点还是蓝链表为头节点
- 保证代码逻辑一致，要不然需要考虑head和非head的情况
- return的时候会很方便
- 不需要检查待合并的链表是否为空
在虚拟头节点初始化cur，方便我们遍历两个链表
每次对list1和list2的val进行判断，哪个比较小，cur.next就赋值哪个
在赋值完之后记得移动cur指针，否则最后return的时候只会有一个节点

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def mergeTwoLists(self, list1: Optional[ListNode], list2: Optional[ListNode]) -> Optional[ListNode]:
        head = ListNode()
        current = head

        while list1 and list2:
            if list1.val < list2.val:
                current.next = list1
                list1 = list1.next
            else:
                current.next = list2
                list2 = list2.next
            # 赋值完之后移动指针
            current = current.next
        
        #有一条较短的链表已经遍历完，直接把长链表剩下的部分接上就好了
        current.next = list1 or list2
        return head.next

一些关于return a or b和return a and b的细节：
chatGPT
在这里插入图片描述

复杂度分析

时间复杂度：O(a+b)
空间复杂度：O(1)

143. 重排链表

Leetcode
我当时写的时候觉得很好的一道题，因为结合了翻转和合并，还用到了快慢针来确定链表中值。

思路

重排的思路是将后半部分的链表翻转，然后将其与前半部分的链表合并。
快慢指针判断中点位置，因为快针一次走两步，慢针一次一步，当快针遍历到尾部的时候，慢针所在位置是中点。

如图所示，快慢指针初始都从head出发，while loop运行到fast不能再前进为止（fast每次走两个节点）。这种写法可以保证不论链表的长度奇偶，代码的逻辑都是一致。
翻转的思路是用双指针prev和cur
合并的思路也是用双指针，但是与上一题有一些不一样

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def reorderList(self, head: Optional[ListNode]) -> None:
        """
        Do not return anything, modify head in-place instead.
        """

        fast, slow = head, head

        # slow 的位置是中点
        while fast.next and fast.next.next:
            fast = fast.next.next
            slow = slow.next

        prev, cur = None, slow.next
        slow.next = None

        #翻转后半部分，经过翻转后的链表头为prev
        while cur:
            temp = cur.next
            cur.next = prev
            prev = cur
            cur = temp

        #合并
        main, sub = head, prev
        while sub:
            nextNode = main.next
            main.next = sub
            main = sub
            sub = nextNode

复杂度分析

时间复杂度：找中点，翻转，合并的时间复杂度都是O(n)，所以总体的复杂度也是O(n)
空间复杂度：O(1)

思路二

还有另外一个比较简单的思路就是用到数组和双指针。利用数组的首尾双指针来定位需要连接的链表节点。

class Solution:
    def reorderList(self, head: ListNode) -> None:
        if not head:
            return
        
        vec = list()
        node = head
        while node:
            vec.append(node)
            node = node.next
        
        i, j = 0, len(vec) - 1
        while i < j:
            vec[i].next = vec[j]
            i += 1
            if i == j:
                break
            vec[j].next = vec[i]
            j -= 1
        
        vec[i].next = None

复杂度分析

时间复杂度：O(n)
空间复杂度：O(n)，因为用到了数组来存储链表节点

enzoherewj

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
1
评论
代码随想录算法训练营第四天 | 链表 part 2

快慢指针判断中点位置，因为快针一次走两步，慢针一次一步，当快针遍历到尾部的时候，慢针所在位置是中点。新链表是通过拼接给定的两个链表的所有节点组成的。我当时写的时候觉得很好的一道题，因为结合了翻转和合并，还用到了快慢针来确定链表中值。这道题可以像卡哥那样，先判断两条链表的长度，求差值，再末端对齐，开始。如果链表无环，则返回。重排的思路是将后半部分的链表翻转，然后将其与前半部分的链表合并。你不能只是单纯的改变节点内部的值，而是需要实际的进行节点交换。给定一个链表，两两交换其中相邻的节点，并返回交换后的链表。
复制链接

扫一扫