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

Python에서 k 크기의 그룹으로 연결 목록을 반전시키는 프로그램

<시간/>

단일 연결 목록이 있고 또 다른 값 k가 있다고 가정하면 모든 k개의 인접한 노드 그룹을 반대로 해야 합니다.

따라서 입력이 List =[1,2,3,4,5,6,7,8,9,10], k =3인 경우 출력은 [3, 2, 1, 6, 5 , 4, 9, 8, 7, 10, ]

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

  • tmp :=값이 0인 새 노드
  • tmp 다음:=노드
  • 이전 :=null, curr :=null
  • lp :=임시, lc :=현재
  • cnt :=k
  • curr이 null이 아닌 동안 do
    • 이전 :=null
    • cnt> 0이고 curr이 null이 아닌 동안 do
      • 다음 :=curr의 다음
      • curr의 다음:=이전
      • 이전 :=curr, curr :=다음
      • cnt :=cnt - 1
    • lp 다음 :=prev, lc 다음 :=curr
    • lp :=lc, lc :=curr
    • cnt :=k
  • tmp 다음에 반환

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

예시

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, k):
      tmp = ListNode(0)
      tmp.next = node
      prev, curr = None, node
      lp, lc = tmp, curr
      cnt = k
      while curr:
         prev = None
         while cnt > 0 and curr:
            following = curr.next
            curr.next = prev
            prev, curr = curr, following
            cnt -= 1
         lp.next, lc.next = prev, curr
         lp, lc = lc, curr
         cnt = k
      return tmp.next
ob = Solution()
head = make_list([1,2,3,4,5,6,7,8,9,10])
print_list(ob.solve(head, 3))

입력

[1,2,3,4,5,6,7,8,9,10], 3

출력

[3, 2, 1, 6, 5, 4, 9, 8, 7, 10, ]