Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

กลับลิงค์ลิสต์ใน Python

สมมติว่าเรามีลิงค์ลิสต์ เราต้องกลับมัน เช่นถ้าลิสต์เป็น 1 → 3 → 5 → 7 ลิสต์ที่กลับแล้วจะเป็น 7 → 5 → 3 → 1

วิธีการแก้ปัญหา

เราจะใช้แนวทางนี้:

  • กำหนดกระบวนการหนึ่งเพื่อทำการกลับลิสต์แบบเรียกซ้ำเป็น solve(head, back)
  • ถ้า head ไม่มีอยู่ ให้คืนค่า head
  • temp := head.next
  • head.next := back
  • back = head
  • ถ้า temp ว่าง ให้คืนค่า head
  • head = temp
  • คืนค่า solve(head, back)

ตัวอย่าง

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(object):
    def reverseList(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        return self.solve(head, None)

    def solve(self, head, back):
        if not head:
            return head
        temp = head.next
        head.next = back
        back = head
        if not temp:
            return head
        head = temp
        return self.solve(head, back)

list1 = make_list([1,3,5,7])
ob1 = Solution()
list2 = ob1.reverseList(list1)
print_list(list2)

อินพุต

list1 = [1,3,5,7]

เอาต์พุต

[7, 5, 3, 1, ]