欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

LeetCode——移除链表元素

程序员文章站 2024-01-20 16:05:58
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著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。解题思路:因为我们需...

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