欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Swap Nodes in Pairs

程序员文章站 2022-03-05 08:12:35
...

medium, linked list

Question

交换链列的两个相邻节点,返回新链列的头。

Example
输入: 1->2->3->4
返回:2->1->4->3

Space complexity O(1)。节点值不能改变,只有节点本身可以改变

Solution

假设链列的当前节点为p, 紧接的两个节点是q, r。则第一对交换后:
p->q->r->s 变为q->p->r->s
这里需要判断下一对节点是否存在,如果存在,p的下一个节点就是s。
在交换下一对节点前需要记录当前节点,我们需要判断是接下来否包含两个或更多节点。这个依然可以借助dummyHead来简化问题

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def swapPairs(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        dummy = ListNode(0)
        dummy.next = head
        p = head
        pre = dummy
        while p != None and p.next != None:
            q = p.next
            r = p.next.next
            pre.next = q
            q.next = p
            p.next = r
            pre = p
            p = r
        return dummy.next