算法题之反转链表

反转链表

给你单链表的头节点 head ，请你反转链表，并返回反转后的链表。

示例 1：

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

示例 2：

输入：head = [1,2]
输出：[2,1]

示例 3：

输入：head = []
输出：[]

提示：

• 链表中节点的数目范围是 [0, 5000]
• -5000 <= Node.val <= 5000

进阶：链表可以选用迭代或递归方式完成反转。你能否用两种方法解决这道题？

解题思路一：迭代

我们可以定义两个链表节点变量，用pre来记录前一个节点，用cur来记录当前节点。每次我们向后遍历一个节点，并且把当前节点的下一个节点设置为前一个节点；然后pre和cur不断地向后遍历一直到最后一个节点。代码如下：

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode reverseList(ListNode head) {

        if(head == null || head.next == null)return head;

        ListNode pre = null;
        ListNode cur = head;

        while(cur != null){
            ListNode next = cur.next;

            cur.next = pre;
            pre = cur;
            cur = next;
        }

        return pre;
    }
}

复杂度分析

• 时间复杂度：O(n)，需要遍历一遍链表。
• 空间复杂度：O(1)，只需要常量的空间存储变量。

解题思路二：递归

我们可以调用自身函数，参数就是当前节点的下一个节点，即head.next，这样我们相当于不断地递归到了最后一个节点，然后我们返回这个节点，也就是反转链表以后的头节点。接下来，我们需要把当前节点下一个节点的下一个节点设置为当前节点，即完成反转；并且要断开原来的正常的节点关系，即把当前节点的下一个节点设置为null。

具体代码如下：

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode reverseList(ListNode head) {

        if(head == null || head.next == null)return head;

        ListNode node = reverseList(head.next);
        head.next.next = head;
        head.next = null;

        return node;
    }
}

复杂度分析

• 时间复杂度：O(n)，需要遍历一遍链表。
• 空间复杂度：O(n)，除了常量的空间，因为使用了栈空间，而栈空间的深度为n层。