反转一个单链表。
示例:
输入: 1->2->3->4->5->NULL 输出: 5->4->3->2->1->NULL
进阶:
你可以迭代或递归地反转链表。你能否用两种方法解决这道题?
这是一道非常简单的题,首先考虑原地非递归反转。用头插法即可。
/**
* 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) {
if(head == NULL || head->next == NULL) {
return head;
}
ListNode s(0);
for(ListNode * p = head; p != NULL;) {
ListNode * next = p->next;
p->next = s.next;
s.next = p;
p = next;
}
return s.next;
}
};
考虑递归原地反转。只用将上面的for循环修改为递归即可。如果每次保留当前节点前面部分的逆置链表,并将当前节点设置为逆置链表的头节点,那么很容易就可以得到一个逆置链表。
void reverseListAux(ListNode* head, ListNode* &rest) {
if(head == NULL) {
return;
}
ListNode * next = head->next;
head->next = rest;
rest = head;
reverseListAux(next, rest);
}
ListNode* reverseList(ListNode* head) {
if(head == NULL || head->next == NULL) {
return head;
}
ListNode * rest = NULL;
reverseListAux(head, rest);
return rest;
}