มี Linked List และ i, j กลับลำดับโหนดจาก i ถึง j คืน head ใหม่
เช่น [1,2,3,4,5,6,7,8,9] i=2 j=6 ให้ [1,2,7,6,5,4,3,8,9]
ขั้นตอนการแก้ปัญหา
- สร้าง dummy node
prev_headชี้ head - เดินไปหาโหนดก่อน i (
rev_before) และโหนด i (rev_end) - กลับโหนดจาก i ถึง j แบบ iterative (3 pointers)
- เชื่อม
rev_before.next = new_head, rev_end.next = after_j
ตัวอย่างโค้ด
class ListNode:
def __init__(self, data, next=None):
self.val = data; self.next = next
def make_list(elements):
head = ListNode(elements[0])
for e in elements[1:]:
ptr = head
while ptr.next: ptr = ptr.next
ptr.next = ListNode(e)
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, i, j):
prev_head = ListNode(None, node)
prev, curr = prev_head, node
for _ in range(i): prev, curr = curr, curr.next
rev_before, rev_end = prev, curr
for _ in range(j-i+1):
tmp = curr.next; curr.next = prev; prev, curr = curr, tmp
rev_before.next, rev_end.next = prev, curr
return prev_head.next
ob = Solution()
head = make_list([1,2,3,4,5,6,7,8,9])
i=2; j=6
print_list(ob.solve(head, i, j))
อินพุต
[1,2,3,4,5,6,7,8,9], 2, 6
เอาต์พุต
[1, 2, 7, 6, 5, 4, 3, 8, 9, ]
