36. Remove Linked List Elements

문제

val을 인수로 받아 연결 리스트에서 val값만 삭제하시오.

Input: head = [1,2,6,3,4,5,6], val = 6
Output: [1,2,3,4,5]

 

해결 방법

https://leetcode.com/problems/remove-linked-list-elements/discuss/57461/Simple-and-clear-python-solution

def removeElements(head, val):
    # 첫 번째 노드가 val인 경우 head를 다음 노드로 지정
    while head and head.val == val:
            head = head.next
            
    current = head

    while current and current.next:
        if current.next.val == val:
            current.next = current.next.next
        else:
            current = current.next

    return head