第三天 | 203.移除链表元素、707.设计链表、206.反转链表 | 补卡

LeetCode 203. 移除链表元素

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

思路：

class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        # 比较第一个节点
        while head is not None and head.val == val:
            head = head.next

        p = head
        
        # 比较后续节点
        while p != None and p.next != None:
            if p.next.val == val: # 下一个节点等于val
                p.next = p.next.next
            else:
                p = p.next
        return head

LeetCode 707. 设计链表

你可以选择使用单链表或者双链表，设计并实现自己的链表。
单链表中的节点应该具备两个属性：val 和 next 。val 是当前节点的值，next 是指向下一个节点的指针/引用。

如果是双向链表，则还需要属性 prev 以指示链表中的上一个节点。假设链表中的所有节点下标从 0 开始。

实现 MyLinkedList 类：
–MyLinkedList() 初始化 MyLinkedList 对象。
–int get(int index) 获取链表中下标为 index 的节点的值。如果下标无效，则返回 -1 。
–void addAtHead(int val) 将一个值为 val 的节点插入到链表中第一个元素之前。在插入完成后，新节点会成为链表的第一个节点。
–void addAtTail(int val) 将一个值为 val 的节点追加到链表中作为链表的最后一个元素。
–void addAtIndex(int index, int val) 将一个值为 val 的节点插入到链表中下标为 index 的节点之前。如果 index 等于链表的长度，那么该节点会被追加到链表的末尾。如果 index 比长度更大，该节点将 不会插入 到链表中。
–void deleteAtIndex(int index) 如果下标有效，则删除链表中下标为 index 的节点。

思路：使用指针要特别注意越界问题，使用虚拟头结点，指向链表的头结点，后续操作都是从虚拟头结点出发，尽量不要使用双指针，容易越界。

class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

class MyLinkedList:

    def __init__(self):
        # 初始化自己的链表,创建一个虚拟头节点
        self.dummy_head = ListNode()
        self.size = 0

    def get(self, index: int) -> int:
        if index < 0 or index >= self.size:
            return -1
        p = self.dummy_head.next
        for i in range(index):
            p = p.next
        return p.val

    def addAtHead(self, val: int) -> None:
        new_head = ListNode(val, self.dummy_head.next)
        self.dummy_head.next = new_head
        self.size += 1


    def addAtTail(self, val: int) -> None:
        new_tail = ListNode(val, None)
        if self.dummy_head == None:
            self.head = new_tail
        p = self.dummy_head
        
        while p.next != None:
            p = p.next
        p.next = new_tail
        self.size += 1

    def addAtIndex(self, index: int, val: int) -> None:
        if index < 0 or index > self.size:
            return 
            
        p = self.dummy_head
        for i in range(index):
            p = p.next
           
        p.next = ListNode(val, p.next)
        self.size += 1

    def deleteAtIndex(self, index: int) -> None:
        if index < 0 or index >= self.size:
            return
            
        p = self.dummy_head
        for i in range(index):
            p = p.next

        p.next = p.next.next
        self.size -= 1

# Your MyLinkedList object will be instantiated and called as such:
# obj = MyLinkedList()
# param_1 = obj.get(index)
# obj.addAtHead(val)
# obj.addAtTail(val)
# obj.addAtIndex(index,val)
# obj.deleteAtIndex(index)

LeetCode 707. 设计链表

给你单链表的头节点 head ，请你反转链表，并返回反转后的链表。

思路：双指针法，创建p，q指针分别代表前一个结点和当前结点，再创建一个temp用于储存p.next。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def reverseList(self, head: Optional[ListNode]) -> Optional[ListNode]:
        p = None
        q = head
        
        while q != None:
            temp = q.next
            q.next = p
            p = q
            q = temp

        return p