Swap Nodes in Pairs

2017-07-28  本文已影响0人  穿越那片海

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
上一篇下一篇

猜你喜欢

热点阅读