題目:輸入一個鏈表,反轉鏈表後,輸出鏈表的所有元素。
Java代碼如下(主要思路見註解):
鏈接:https://www.nowcoder.com/questionTerminal/75e878df47f24fdc9dc3e400ec6058ca
來源:牛客網
public class Solution {
public ListNode ReverseList(ListNode head) {
if(head==null)
return null;
//head爲當前節點,如果當前節點爲空的話,那就什麼也不做,直接返回null;
ListNode pre = null;
ListNode next = null;
//當前節點是head,pre爲當前節點的前一節點,next爲當前節點的下一節點
//需要pre和next的目的是讓當前節點從pre->head->next1->next2變成pre<-head next1->next2
//即pre讓節點可以反轉所指方向,但反轉之後如果不用next節點保存next1節點的話,此單鏈表就此斷開了
//所以需要用到pre和next兩個節點
//1->2->3->4->5
//1<-2<-3 4->5
while(head!=null){
//做循環,如果當前節點不爲空的話,始終執行此循環,此循環的目的就是讓當前節點從指向next到指向pre
//如此就可以做到反轉鏈表的效果
//先用next保存head的下一個節點的信息,保證單鏈表不會因爲失去head節點的原next節點而就此斷裂
next = head.next;
//保存完next,就可以讓head從指向next變成指向pre了,代碼如下
head.next = pre;
//head指向pre後,就繼續依次反轉下一個節點
//讓pre,head,next依次向後移動一個節點,繼續下一次的指針反轉
pre = head;
head = next;
}
//如果head爲null的時候,pre就爲最後一個節點了,但是鏈表已經反轉完畢,pre就是反轉後鏈表的第一個節點
//直接輸出pre就是我們想要得到的反轉後的鏈表
return pre;
}
}
同樣的思路,C++代碼如下:
class Solution{
public:
ListNode* ReverseList(ListNode* pHead){
if (pHead == NULL){
return NULL;
}
ListNode* pre = NULL;
ListNode* next = NULL;
while (pHead){
next = pHead->next;
pHead->next = pre;
pre = pHead;
pHead = next;
}
return pre;
}
};
或者:
class Solution {
public:
ListNode* ReverseList(ListNode* pHead) {
if (pHead == NULL)return NULL;
ListNode* pre = NULL;
ListNode* cur = pHead;
while (cur) {
ListNode* tmp;
tmp = cur->next;
cur->next = pre;
pre = cur;
cur = tmp;
}
return pre;
}
};