给定一个链表,两两交换其中相邻的节点,并返回交换后的链表。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
示例 1:
输入:head = [1,2,3,4]
输出:[2,1,4,3]
示例 2:
输入:head = []
输出:[]
示例 3:
输入:head = [1]
输出:[1]
提示:
- 链表中节点的数目在范围 [0, 100] 内
- 0 <= Node.val <= 100
进阶:你能在不修改链表节点值的情况下解决这个问题吗?(也就是说,仅修改节点本身。)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/swap-nodes-in-pairs
解题思路
利用stack
我们利用一个 stack,然后不断迭代链表,每次取出两个节点放入 stack 中,再从 stack 中拿出两个节点。
借助 stack 后进先出的特点,放进去的时候是 1,2 。拿出来的时候就是 2,1 两个节点了。
再把这两个节点串联起来,重复这个逻辑遍历完整个链表,就可以做到两两反转的效果了。
虽然用到了 stack,但因为只存了两个元素,所以空间复杂度还是 O(1)O(1),时间复杂度是 O(n)O(n)。
代码实现:
class Solution(object):
def swapPairs(self, head):
if not (head and head.next):
return head
p = ListNode(-1)
# 用stack保存每次迭代的两个节点
# head指向新的p节点,函数结束时返回head.next即可
cur,head,stack = head,p,[]
while cur and cur.next:
# 将两个节点放入stack中
_,_ = stack.append(cur),stack.append(cur.next)
# 当前节点往前走两步
cur = cur.next.next
# 从stack中弹出两个节点,然后用p节点指向新弹出的两个节点
p.next = stack.pop()
p.next.next = stack.pop()
p = p.next.next
# 注意边界条件,当链表长度是奇数时,cur就不为空
if cur:
p.next = cur
else:
p.next = None
return head.next