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

โปรแกรมค้นหาโหนดสุดท้ายที่ K-th ของรายการที่เชื่อมโยงใน Python


สมมติว่าเรามีรายการที่เชื่อมโยงเพียงอย่างเดียว เราต้องตรวจสอบหาค่าของโหนดสุดท้ายที่ k (ดัชนี 0 ดัชนี) เราต้องแก้ปัญหานี้ในครั้งเดียว

ดังนั้น ถ้าอินพุตเหมือนกับ node =[5,4,6,3,4,7], k =2 เอาต์พุตจะเป็น 3 เนื่องจากโหนดสุดท้าย (ดัชนี 3) มีค่า 3.

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -

  • klast :=โหนด

  • สุดท้าย :=โหนด

  • สำหรับฉันอยู่ในช่วง 0 ถึง k ทำ

    • สุดท้าย :=ถัดไปจากสุดท้าย

  • ในขณะที่ next of last ไม่เป็นโมฆะ ทำ

    • สุดท้าย :=ถัดไปจากสุดท้าย

    • klast :=ถัดไปจาก klast

  • คืนค่าของ klast

ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -

ตัวอย่าง

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

class Solution:
   def solve(self, node, k):
      klast = node
      last = node
      for i in range(k):
         last = last.next
      while last.next:
         last = last.next
         klast = klast.next
      return klast.val

ob = Solution()
l1 = make_list([5,4,6,3,4,7])
print(ob.solve(l1, 2))

อินพุต

[5,4,6,3,4,7], 2

ผลลัพธ์

3