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

本文地址:https://blog.csdn.net/weixin_45295612/article/details/113977743