단일 연결 목록과 하나의 대상이 있다고 가정하고 대상과 값이 동일한 모든 노드를 삭제한 후 동일한 연결을 반환해야 합니다.
따라서 입력이 [5,8,2,6,5,2,9,6,2,4]와 같으면 출력은 [5, 8, 6, 5, 9, 6, 4, ]입니다.
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
- 헤드:=노드
- node와 node.next가 null이 아닌 동안 do
- 다음 노드의 값이 target과 같을 때 do
- 다음 노드 :=다음 노드
- 노드:=노드의 다음
- 다음 노드의 값이 target과 같을 때 do
- head의 값이 target과 같으면
- 머리 다음 반환
- 그렇지 않으면
- 리턴 헤드
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
예시
class ListNode:
def __init__(self, data, next = None):
self.val = data
self.next = next
def make_list(elements):
head = ListNode(elements[0])
for element in elements[1:]:
ptr = head
while ptr.next:
ptr = ptr.next
ptr.next = ListNode(element)
return head
def print_list(head):
ptr = head
print('[', end = "")
while ptr:
print(ptr.val, end = ", ")
ptr = ptr.next
print(']')
class Solution:
def solve(self, node, target):
head = node
while (node and node.next):
while node.next.val == target:
node.next = node.next.next
node = node.next
if head.val == target:
return head.next
else:
return head
ob = Solution()
head = make_list([5,8,2,6,5,2,9,6,2,4])
ob.solve(head, 2)
print_list(head) 입력
[5,8,2,6,5,2,9,6,2,4]
출력
[5, 8, 6, 5, 9, 6, 4, ]