Leetcode题库——24.两两交换链表中的节点

时间:2023-03-08 23:30:26
Leetcode题库——24.两两交换链表中的节点

@author: ZZQ

@software: PyCharm

@file: swapPairs.py

@time: 2018/10/20 19:49

说明:给定一个链表,两两交换其中相邻的节点,并返回交换后的链表。

示例:

给定 1->2->3->4, 你应该返回 2->1->4->3.

说明:

你的算法只能使用常数的额外空间。

你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。

思路:

四个节点,分别记录当前需要进行交换的两个节点(first, second),以及这俩个节点的前后节点(pre, post)

然后每次只针对这四个节点进行交换即可。

注意考虑当输入是空节点,一个节点,两个节点,三个节点以及节点个数为奇数的情况。

class ListNode(object):
def __init__(self, x):
self.val = x
self.next = None class Solution(object):
def __init__(self):
pass def exchange(self, pre, first, second, post):
first.next = None
second.next = None
first.next = post
second.next = first
pre.next = second
def swapPairs(self, head):
"""
:type head: ListNode
:rtype: ListNode
"""
if head is None or head.next is None:
return head
pre = ListNode(0)
pre.next = head
first = head
second = head.next
post = second.next
p = pre
while True:
self.exchange(pre, first, second, post)
pre = pre.next.next
first = pre.next
if first is None:
break
second = pre.next.next
if second is None:
break
post = post.next.next
if post is None:
self.exchange(pre, first, second, post)
break
return p.next if __name__ == "__main__":
answer = Solution()
l1 = ListNode(1)
p1 = ListNode(2)
p2 = ListNode(3)
p3 = ListNode(4)
p4 = ListNode(5)
p5 = ListNode(6)
l1.next = p1
p1.next = p2
p2.next = p3
p3.next = p4
p4.next = p5 l2 = answer.swapPairs(l1)
while l2 is not None:
print l2.val
l2 = l2.next