328. Odd Even Linked List

개굴·2024년 6월 22일

leetcode

목록 보기
37/51
  • python3

Problem

Given the head of a singly linked list, group all the nodes with odd indices together followed by the nodes with even indices, and return the reordered list.

The first node is considered odd, and the second node is even, and so on.

Note that the relative order inside both the even and odd groups should remain as it was in the input.

You must solve the problem in O(1) extra space complexity and O(n) time complexity.

Example 1:

Input: head = [1,2,3,4,5]
Output: [1,3,5,2,4]

Example 2:

Input: head = [2,1,3,5,6,4,7]
Output: [2,3,6,7,1,5,4]

Constraints:

  • The number of nodes in the linked list is in the range [0, 104].
  • -106 <= Node.val <= 106

Pseudocode

  1. Initialize the odd pointer to the first node and the even pointer to the second node, then store even_head to later connect it to the end of the odd list.

  2. While even and even.next exits, repeat the following:

    • set odd.next to even.next to point to the next odd node.
    • move odd to the next odd node.
    • set even.next to odd.next to point to the next even node.
    • move even to the next even node.
  3. Once the loop is finished, set odd.next to even_head to connect the odd and even lists.

Code

# 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: Optional[ListNode]) -> Optional[ListNode]:
        
        if not head:
            return head

        odd = head
        even = head.next
        even_head = even

        while even and even.next:
            odd.next = even.next
            odd = odd.next
            even.next = odd.next
            even = even.next

        odd.next = even_head
        return head

Result

  • Time Complexity : O(n)
  • Space Complexity : O(1)
profile
알쏭달쏭혀요

0개의 댓글