移除链表元素-----python

题目描述

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

示例 1：

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

输入：head = [], val = 1
输出：[]
示例 3：

输入：head = [7,7,7,7], val = 7
输出：[]

提示：

列表中的节点数目在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= val <= 50

法一，有头链表：

# 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
        """
        head1=ListNode()
        head1.next=head
        temp=head1
        while temp.next:
            if temp.next.val == val:
                temp.next=temp.next.next
            else:
                temp=temp.next
        return head1.next
            
            

思路如下：遍历所有的链表。首先建立一个空节点，并使空节点指向输入的链表。并用一个暂存符接收头节点。进行循环遍历。相同跳过，不同则指向下一个。

法二，无头链表：

# 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
        """
#检查头节点是否等于该值
        while head and head.val == val:
            head = head.next
#判断链表是否为空
        if not head:
            return None
        curr=head
#用curr接head的值，从第二个开始遍历即可。
        while curr.next:
            if curr.next.val == val:
                curr.next=curr.next.next
            else:
                curr = curr.next
        return head

心得：检查头节点的方法需要记住。