leetcode算法

leetcode链表之两两交换链表中的结点

2022-03-21  本文已影响0人  小奚有话说

24、两两交换链表中的节点

题目

给你一个链表,两两交换其中相邻的节点,并返回交换后链表的头节点。你必须在不修改节点内部的值的情况下完成本题(即,只能进行节点交换)。

示例1:

输入:head = [1,2,3,4]
输出:[2,1,4,3]

示例2:

输入:head = []
输出:[]

示例3:

输入:head = [1]
输出:[1]

思路

既然题目要求两两交换,那么每次步长为2,如果存在那么交换两个结点的值即可

代码:

class Solution:
    def swapPairs1(self, head: ListNode) -> ListNode:
        if not head: return head
        vHead = ListNode(0, head)
        cur = vHead
        # 判断下个结点,以及下下个结点是否存在,如果存在则将下个结点和下下个结点进行交换
        # 如果不存在,则说明不能两两交换,则循环结束
        while cur.next and cur.next.next:
            # 结点顺序 cur node1 node2
            node1 = cur.next
            node2 = cur.next.next
            # 第一步将node1的下一个结点指向node2的下一个结点
            node1.next = node2.next
            # 第二步将node2的下个结点指向node1
            node2.next = node1
            # 在将cur的下一个结点指向node2,此时就完成node1和node2结点的交换
            cur.next = node2
            # 交换后的节点顺序cur node2 node1,下一次循环则需要将cur移动到node1的位置
            cur = node1
        return vHead.next
上一篇 下一篇

猜你喜欢

热点阅读