面试题——反转链表

AcWing 35 反转链表

题目

定义一个函数，输入一个链表的头结点，反转该链表并输出反转后链表的头结点。

思考题：

• 请同时实现迭代版本和递归版本。

样例

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

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

思路

迭代版本

​ 这道题使用迭代的思想很容易思考，我们只需要准备三个指针，分别是上一个节点prev、当前节点curr和curr的下一个节点node

​ 通过遍历整个链表，没到一个节点，我们先保存curr.next给node，再将curr.next赋给prev，也就改变了curr的指向，随后我们把curr和prev都后移一位即可（在原来的顺序上，因为我们已经保留了node），当curr == null时，说明链表已经到结尾了，我们返回prev即可

递归版本

​ 递归版本如何思考呢，我们只需要思考一种情况，由于是递归实现，当递归到head节点时，我们需要递归函数返回的结果是一个从head.next开始已经被反转的链表，所以我们只需要将head接到链表的结尾，并将head.next指空即可

代码

c++ 迭代

class Solution {
public:
	ListNode* reverseList(ListNode* head) {
		ListNode *prev = NULL, *cur = head;
		while (cur) {
			auto node = cur->next;
			cur->next = prev;
			prev = cur;
			cur = node;
		}
		return prev;
	}
};

c++ 递归

class Solution {
public:
	ListNode* reverseList(ListNode* head) {
		if (!head || !head->next)
			return head;
		auto tail = reverseList(head->next);
		head->next->next = head;
		head->next = NULL;
		return tail;
	}
};

java 迭代

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

java 递归

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