以 Zig-Zag 方式重新排列链表

介绍

本文介绍了如何以 Zig-Zag 方式重新排列链表。

Zig-Zag 排列的原理是将链表分成两个部分，一个部分按照正向顺序排列，另一个部分按照反向顺序排列，最后将两部分链表依次拼接即可。

思路

1. 将原链表按照正向顺序遍历，将元素按照顺序插入正向链表中；
2. 将原链表按照反向顺序遍历，将元素按照顺序插入反向链表中；
3. 将正向链表的尾指针指向反向链表的头指针，反向链表的尾指针指向正向链表的头指针；
4. 将两个链表连接起来即可。

代码实现

以下为 Python 代码片段：

class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next

def zigzag(head: ListNode) -> ListNode:
    if not head or not head.next:
        return head
    
    dummy1 = ListNode(0)
    dummy2 = ListNode(0)
    p1, p2 = dummy1, dummy2
    flag = True
    
    while head:
        if flag:
            p1.next = head
            p1 = p1.next
        else:
            p2.next = head
            p2 = p2.next
        flag = not flag
        head = head.next
    
    p2.next = None
    
    cur1, cur2 = dummy1.next, dummy2.next
    dummy = ListNode(0)
    tail = dummy
    
    while cur1 and cur2:
        tail.next = cur1
        cur1 = cur1.next
        tail.next.next = cur2
        cur2 = cur2.next
        tail = tail.next.next
    
    if cur1:
        tail.next = cur1
    else:
        tail.next = cur2
    
    return dummy.next

总结

本文介绍了 Zig-Zag 排列链表的基本原理和实现方法，希望能对大家的编程工作有所帮助。