算法训练 | leetcode 203. 移除链表元素 707. 设计链表 206. 反转链表

203. 移除链表元素

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

示例:输入：head = [1,2,6,3,4,5,6], val = 6         输出：[1,2,3,4,5]

python3 : 虚拟头节点

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        dummy_head = ListNode(next = head)
        cur = dummy_head
        while cur.next!=None:
            if cur.next.val == val:
                cur.next =  cur.next.next
            else:
                cur= cur.next
        return dummy_head.next

707. 设计链表

设计链表的实现。选择使用单链表或双链表。单链表中的节点应该具有两个属性：val 和 next。val 是当前节点的值，next 是指向下一个节点的指针/引用。如果要使用双向链表，则还需要一个属性 prev 以指示链表中的上一个节点。假设链表中的所有节点都是 0-index 的。

在链表类中实现这些功能：

get(index)：获取链表中第 index 个节点的值。如果索引无效，则返回-1。
addAtHead(val)：在链表的第一个元素之前添加一个值为 val 的节点。插入后，新节点将成为链表的第一个节点。
addAtTail(val)：将值为 val 的节点追加到链表的最后一个元素。
addAtIndex(index,val)：在链表中的第 index 个节点之前添加值为 val  的节点。如果 index 等于链表的长度，则该节点将附加到链表的末尾。如果 index 大于链表长度，则不会插入节点。如果index小于0，则在头部插入节点。
deleteAtIndex(index)：如果索引 index 有效，则删除链表中的第 index 个节点。

示例：

MyLinkedList linkedList = new MyLinkedList();
linkedList.addAtHead(1);
linkedList.addAtTail(3);
linkedList.addAtIndex(1,2);   //链表变为1-> 2-> 3
linkedList.get(1);            //返回2
linkedList.deleteAtIndex(1);  //现在链表是1-> 3
linkedList.get(1);            //返回3

python3:注意Node类的定义，初始化中的self.node就是虚拟头节点，一定不能直接操作头节点，应用临时变量cur指向需要的节点，这样我们还能取到头节点。

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

class MyLinkedList:

    def __init__(self):
        self.node = Node()
        self.size = 0


    def get(self, index: int) -> int:
        if index<0 or index>=self.size:
            return -1
        cur = self.node.next
        while index:
            cur = cur.next
            index-=1
        return cur.val

    def addAtHead(self, val: int) -> None:
        head = Node(val)
        cur = self.node
        head.next = cur.next
        cur.next = head
        self.size+=1



    def addAtTail(self, val: int) -> None:
        tail = Node(val)
        cur = self.node
        while cur.next != None:
            cur = cur.next
        tail.next = cur.next
        cur.next = tail
        self.size+=1


    def addAtIndex(self, index: int, val: int) -> None:
        middle = Node(val)
        cur = self.node
        if index<0:
            index = 0
        if index>self.size:
            return None
        while index:
            cur = cur.next
            index-=1
        middle.next = cur.next
        cur.next = middle
        self.size+=1


    def deleteAtIndex(self, index: int) -> None:
        if index<0 or index>=self.size:
            return None
        cur = self.node
        while index:
            cur = cur.next
            index-=1
        cur.next = cur.next.next
        self.size-=1

206. 反转链表

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

示例：   输入：head = [1,2,3,4,5] 输出：[5,4,3,2,1]

python3:双指针方式，一定要将cur.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]:
        prev = None
        cur = head
        while cur:
            temp = cur.next
            cur.next = prev
            prev = cur 
            cur = temp
        return prev

python3:递归法

# 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]:
        def reverse(prev, cur):
            if not cur:
                return prev
            temp = cur.next
            cur.next = prev
            return reverse(cur, temp)
        return reverse(None, head)

题目来源：力扣（LeetCode）