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