2.1 移除链表元素

移除链表元素

1 题目

203题

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

head包含第一个数值

2 解法

2.1 分类解法

class Solution(object):
    def removeElements(self, head, val):
        #将头节点设置为不等于val
        while(head and head.val == val):
            head = head.next
        #返回空链表
        if(head == None):
            return None
        current = head
        while(current.next):
            if(current.next.val == val):
                current.next = current.next.next
            else:
                current = current.next
        return head

2.2 虚拟头节点

在第一种方法，单链表中移除头结点 和 移除其他节点的操作方式是不一样，其实在写代码的时候也会发现，需要单独写一段逻辑来处理移除头结点的情况。

可以设置一个虚拟头结点，这样原链表的所有节点就都可以按照统一的方式进行移除了

class Solution(object):
    def removeElements(self, head, val):
        newhead = ListNode(1, head)
        current = newhead
        while(current.next):
            if(current.next.val == val):
                current.next = current.next.next
            else:
                current = current.next
        return newhead.next