15.反轉鏈表【代碼的魯棒性】

       題目:輸入一個鏈表,反轉鏈表後,輸出鏈表的所有元素。

       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;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章