代码随想录算法训练营第三天（py） | 203.移除链表元素、203.设计链表、206. 翻转链表

链表理论基础

定义一个链表结点：

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

删除B结点：让A结点指向C结点
插入B结点：让A结点指向B结点，让B结点指向C结点

203.移除链表元素（easy）

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

思路

删除结点，即令待删除结点的前一结点指向后一结点。但当要删除首结点时，就没有前一结点。因此不妨在头结点前面加一个空的虚拟头结点，这样就不用单独做一个头结点删除的逻辑了。

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution(object):
    def removeElements(self, head, val):
        """
        :type head: ListNode
        :type val: int
        :rtype: ListNode
        """
        fake_head = ListNode(next = head) 
        # 设置一个虚拟头结点指向真的头结点
        current = fake_head
        while current.next: # 当不是尾结点
            if current.next.val==val: # 如果下一结点是要找的
                current.next = current.next.next # 删除下一结点
            else:
                current = current.next
        return fake_head.next

203.设计链表（normal）－这是对链表的基础题目

力扣链接
你可以选择使用单链表或者双链表，设计并实现自己的链表。
单链表中的节点应该具备两个属性：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(object):

    def __init__(self):
        self.fake_head = ListNode() 
        # 设定虚拟头结点
        self.size = 0

    def get(self, index):  # 查询指定下标
        """
        :type index: int
        :rtype: int
        """
        if index<0 or index >= self.size:
            return -1 # 下标无效，返回－1
        current = self.fake_head.next # 从头结点开始
        for i in range(index):
            current = current.next

        return current.val

    def addAtHead(self, val): # 头插
        """
        :type val: int
        :rtype: None
        """
        self.fake_head.next = ListNode(val, self.fake_head.next)
        # 一行搞定。创建一个指向头结点的结点并赋值，然后让虚拟头结点指向他
        self.size += 1

    def addAtTail(self, val):  # 尾插
        """
        :type val: int
        :rtype: None
        """
        current = self.fake_head
        for i in range(self.size):
        # while current.next:
            current = current.next
        # 循环结束后current是尾结点
        current.next = ListNode(val) # 尾插的结点不用next了，其默认为None
        self.size += 1

    def addAtIndex(self, index, val):
        """
        :type index: int
        :type val: int
        :rtype: None
        """
        if index < 0 or index > self.size: # 这里记得不要>=
            return 
        current = self.fake_head
        for i in range(index):
            current = current.next
        current.next = ListNode(val, current.next)
        self.size += 1

    def deleteAtIndex(self, index):
        """
        :type index: int
        :rtype: None
        """
        if index < 0 or index >= self.size:
            return
        current = self.fake_head
        for i in range(index):
            current = current.next
        current.next = current.next.next # 少推进一位，删除下一个结点
        self.size -= 1

206. 翻转链表

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

思路1 双指针法

定义两个指针，cur指向当前结点，初始化为头结点；prev指向要被cur指向的结点（即前一个结点），初始化为None。步骤如下：

1. 用一个temp保存cur的下一个结点
2. 令cur指向prev
3. 当前结点翻转完成，开始移动指针。prev移动到cur，然后再将cur移动到temp，这里千万不能弄反顺序。
4. 最后返回prev

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution(object):
    def reverseList(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        cur = head
        prev = None
        while cur:
            temp = cur.next
            cur.next = prev
            prev = cur
            cur = temp
        return prev

最后，prev指向原来的尾结点/现在的头结点，cur指向了None。现在prev已经成为新链表的头结点，返回头结点即可获取整个链表。