Question
反转一个单链表。
示例:
输入: 1->2->3->4->5->NULL
输出: 5->4->3->2->1->NULL
进阶:
你可以迭代或递归地反转链表。你能否用两种方法解决这道题?
Answer
第一次提交:32ms/8.3MB
看到题目纯个人思考写出来的代码,主要逻辑如下:
从链表头一直往后读,直到读到末节点返回,然后将最后一个节点当作新链表的头节点,同时将当前节点(head)作为返回的新链表(tmp)的末节点。
问题:冗余遍历,tmp从头到尾遍历了多次,没有利用head->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) {
ListNode *tmp=NULL;
if(NULL == head)return head;
if(NULL != head->next){//如果是末节点直接返回
tmp=reverseList(head->next);
head->next=NULL;
ListNode *tmp2=tmp;
while(tmp2->next != NULL){//读到新链表的最后一个节点
tmp2=tmp2->next;
}
tmp2->next=head;
return tmp;
}
return head;
}
};
第二次提交:8ms/8.3MB
简化了第一次提交的代码,通过利用head->next,直接将新链表最后一个节点(head->next)指向当前节点(head),基本思路还是没变。
ListNode* reverseList(ListNode* head) {
ListNode *tmp=NULL,*result=NULL;
if(NULL == head || NULL == head->next)return head;
result=reverseList(head->next);//因为思路是先遍历到最后一个节点,所以先递再归
tmp=head->next;
head->next=NULL;
tmp->next=head;
return result;
}
第三次提交:8ms/8MB
尝试用迭代做
主要逻辑:result存放新链表的头节点,每向后读一个节点就作为新链表的(新)头节点。
这种逻辑是没有办法用递归做的,因为递归的函数只能传递一个参数ListNode,也就是head->next,而这个逻辑需要能同时知道head->next和head
ListNode* reverseList(ListNode* head) {
ListNode *tmp=NULL,*result=NULL;
if(NULL == head || NULL == head->next)return head;
while(head != NULL)
{
tmp=result;
result=head;
head=head->next;
result->next=tmp;
}
return result;
}