LeetCode206 反转链表

反转一个单链表。

示例:

输入: 1->2->3->4->5->NULL
输出: 5->4->3->2->1->NULL
进阶:
你可以迭代或递归地反转链表。你能否用两种方法解决这道题？

 

递归：

递归到尾节点，然后往回；每次都是返回尾结点。对于每个递归栈的head，将head->next->next设置为自己，即把下一个的下一个设为自己；再把自己的next设置为nullptr，否则头部的next还是指向原来得下一个，会出错

迭代：

一样的思想，只是要提前保存下一个节点。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* reverseList(ListNode* head) {
        return iterate(head);
    }

    ListNode* iterate(ListNode* head){
        if(head==nullptr)
            return head;
        ListNode* prev=head;
        head=head->next;
        prev->next=nullptr;
        while(head!=nullptr){
            ListNode* save=head->next;
            head->next=prev;
            prev=head;
            head=save;
        }
        return prev;
    }

    ListNode* recursive(ListNode* head){
        if(head==nullptr || head->next==nullptr){
            return head;
        }
        ListNode* ret=recursive(head->next);
        head->next->next=head;
        head->next=nullptr;
        return ret;
    }
};