LeetCode203. Remove Linked List Elements

最新推荐文章于 2024-11-04 16:26:24 发布

原创最新推荐文章于 2024-11-04 16:26:24 发布 · 133 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

179 篇文章

订阅专栏

Python

173 篇文章

订阅专栏

本文介绍了一种从链表中移除特定值的所有节点的方法。通过调整指针位置，跳过目标值，最终实现链表的有效清理。适用于数据结构与算法初学者。

Remove all elements from a linked list of integers that have value val.

Example:

Input:  1->2->6->3->4->5->6, val = 6
Output: 1->2->3->4->5

思路:首先要保证链表不为空.其次,要找到真正的开头(如果head值就是val,head需要后移)

找到开头(head)后时,用p表示当前Node.

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def removeElements(self, head, val):
        """
        :type head: ListNode
        :type val: int
        :rtype: ListNode
        """
        
        if head is None:
            return None
        
        while head.val == val:
            if head.next is None:
                return None
            head = head.next
            
        p = head
        
        while p.next:
            if p.next.val != val:
                p = p.next
            elif p.next.next is None:  #提交时候并不需要判断p.next.next.不报错.
                p.next = None
            else:
                p.next = p.next.next
                
        return head