LeetCode——移除链表元素

LeetCode——移除链表元素

题目描述：
删除链表中等于给定值 val 的所有节点。

示例:

输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-linked-list-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路：
因为我们需要最后操作后链表的头结点，所以我们需要先定义一个节点p让其等于初始头结点。
然后我们遍历每一个节点，找到节点值等于val的节点并删除他。
本题的关键问题在于，头结点的值可能也是val，如果只是单纯循环判断则会出错。
如果我们就采用仅仅复制头结点的方式(用H=head)然后用H进行操作，最后返回head。这样就会导致如果头结点也是我们需要删除的节点就会导致错误。当然我们可以采用循环操作判断第一个不是我们要删除的节点，但是这样有些麻烦。
我采用的方法是：先不管头结点，从head.next开始判断，一直遍历完后面的所有节点，最后再去判断头结点。

python代码：

# 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
        """
        if head is None:
            return None
        p = head
        s = None
        while p.next:
            if p.next.val == val:
                s = p.next
                p.next = p.next.next
                del s
            else:
                p = p.next
        if head.val == val:
            s = head
            head = head.next
            del s
        return head