[LeetCode] 206. 反转链表

传送门：[LeetCode] 206. 反转链表

题目描述

反转一个单链表。

示例 :

输入: 1->2->3->4->5->NULL
输出: 5->4->3->2->1->NULL

进阶:

• 你可以迭代或递归地反转链表。你能否用两种方法解决这道题？

链表定义如下：

public class ListNode {
    int val;
    ListNode next;

    ListNode(int x) {
        val = x;
    }
}

分析与代码

解法一、迭代

• 用一个 pre 指针记录前一个结点，初始为 null。
• 每次循环先用 next 指针记录下一个结点，使当前结点指向前一个结点，更新 pre 为当前结点。
• 最后返回的是 pre 结点，因为结束条件是当前结点为 null，到最后 pre 即为新的头结点。

代码：

class Solution {
    public ListNode reverseList(ListNode head) {
        ListNode pre = null;
        while (head != null) {
            ListNode next = head.next;
            head.next = pre;
            pre = head;
            head = next;
        }
        return pre;
    }
}

解法二、递归

• 我们假设只剩链表的前两个结点未反转，后面的结点都已经反转了，此时反转两个结点的操作就是head.next.next = head;，下一个结点指回当前结点，此时这两个结点就形成一个环了，所以还要断开当前结点指向下一个结点的连接。head.next = null;。
• 当然这是假设后面部分都已经反转完成的情况，我们要先递归进入到最后，使用 newHead 记录后面反转完成后的新头结点，并返回 newHead。
• 递归终止条件为剩下的不足两个结点。

代码：

class Solution {
    public ListNode reverseList(ListNode head) {
        if (head == null || head.next == null) {
            return head;
        }
        ListNode newHead = reverseList(head.next);
        head.next.next = head;
        head.next = null;
        return newHead;
    }
}

小结

迭代法比较简单，递归法很有意思。

---