【LeetCode题解】24_两两交换链表中的节点(Swap-Nodes-in-Pairs)

更多 LeetCode 题解笔记可以访问我的 github

描述

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

示例:

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

说明:

  • 你的算法只能使用常数的额外空间。
  • 你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。

解法一:迭代

思路

这道题的思路其实很直接(改变一对节点的 next 指针),比较难的是该如何进行交换。这里,我首先生成一个虚拟头节点 dummy,这样可以将后面执行的操作统一起来,不用区分是否为链表的头部(head)。接着,借助于两个指针 firstsecond,实现两个节点的“交换”。

Java 实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode swapPairs(ListNode head) {
        ListNode dummy = new ListNode(-1);
        dummy.next = head;
        ListNode curr = dummy;
        while (curr.next != null && curr.next.next != null) {
            ListNode first = curr.next;
            ListNode second = curr.next.next;
            
            // swap two nodes
            first.next = second.next;
            second.next = first;
            curr.next = second;
            
            // update to next iteration
            curr = curr.next.next;
        }
        return dummy.next;
    }
}
// Runtime: 2 ms
// Your runtime beats 100.00 % of java submissions.

Python 实现

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def swapPairs(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        dummy = ListNode(-1)
        dummy.next, curr = head, dummy
        while curr.next and curr.next.next:
            first, second = curr.next, curr.next.next
            
            # swap two nodes
            first.next, second.next, curr.next = second.next, first, second
            
            # update to next iteration
            curr = curr.next.next
        return dummy.next
# Runtime: 32 ms
# Your runtime beats 100.00 % of python3 submissions.

复杂度分析

  • 时间复杂度\(O(n)\),其中 \(n\) 表示链表的长度(节点的数目)。循环需要的次数为 \(\left \lfloor \frac{n}{2} \right \rfloor\),且循环中执行的操作的时间复杂度为 \(O(1)\),因此,总的时间复杂度是 \(O(n)\)
  • 空间复杂度\(O(1)\),只需要存储 4 个节点的引用和一个虚拟头结点。

解法二:递归(不满足空间复杂度要求)

思路

递归的方式也需要改变一对节点的 next 指针,不同的是,递归的方式首先递归到链表的尾部,然后从链表的尾部开始交换节点,一直到链表的头部。

Java 实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode swapPairs(ListNode head) {
        if (head == null || head.next == null) {
            return head;
        }
        ListNode p = head.next;
        head.next = swapPairs(head.next.next);
        p.next = head;
        return p;
    }
}
// Runtime: 2 ms
// Your runtime beats 100.00 % of java submissions.

Python 实现

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def swapPairs(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        if not head or not head.next:
            return head
        p = head.next
        head.next = self.swapPairs(head.next.next)
        p.next = head
        return p

复杂度分析

  • 时间复杂度\(O(n)\),其中 \(n\) 表示链表的长度(节点的数目)。
  • 空间复杂度\(O(n)\),递归调用占用系统栈空间,递归的深度为 \(\left \lfloor \frac{n}{2} \right \rfloor\)
posted @ 2018-11-21 21:50  xugenpeng  阅读(460)  评论(0编辑  收藏  举报