Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

파이썬에서와 같은 값을 가진 연결 리스트의 모든 노드를 제거하는 프로그램

<시간/>

단일 연결 목록과 하나의 대상이 있다고 가정하고 대상과 값이 동일한 모든 노드를 삭제한 후 동일한 연결을 반환해야 합니다.

따라서 입력이 [5,8,2,6,5,2,9,6,2,4]와 같으면 출력은 [5, 8, 6, 5, 9, 6, 4, ]입니다.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

  • 헤드:=노드
  • node와 node.next가 null이 아닌 동안 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, ]