328. 奇偶連結串列

Difficulty: 中等

給定一個單鏈表，把所有的奇數節點和偶數節點分別排在一起。請注意，這裡的奇數節點和偶數節點指的是節點編號的奇偶性，而不是節點的值的奇偶性。

請嘗試使用原地演算法完成。你的演算法的空間複雜度應為 O(1)，時間複雜度應為 O(nodes)，nodes 為節點總數。

示例 1:

輸入: 1->2->3->4->5->NULL
輸出: 1->3->5->2->4->NULL

示例 2:

輸入: 2->1->3->5->6->4->7->NULL 
輸出: 2->3->6->7->1->5->4->NULL
 

說明:

• 應當保持奇數節點和偶數節點的相對順序。
• 連結串列的第一個節點視為奇數節點，第二個節點視為偶數節點，以此類推。

Solution

# Definition for singly-linked list.
# class ListNode:
#   def __init__(self, val=0, next=None):
#     self.val = val
#     self.next = next
class Solution:
  def oddEvenList(self, head: ListNode) -> ListNode:
    if not head or not head.next: 
      return head
    
    p1 = head
    p2 = head.next
    pre = p2
    
    while p2 and p2.next:
      p1.next = p2.next
      p1 = p1.next
      p2.next = p1.next
      p2 = p2.next
    
    p1.next = pre
    return head