1. 程式人生 > >Lintcode 35.翻轉連結串列

Lintcode 35.翻轉連結串列

題目描述

思路 剛開始時沒注意這個問題時不帶頭結點的,即head指向第一個元素

基本思路還是從第二個元素依次往前加

但是要注意特殊情況 — 連結串列為空或者只有一個元素的時候特殊處理

程式碼如下

#include <iostream>
using namespace std;
/**
 * Definition of singly-linked-list:
 *
 * class ListNode {
 * public:
 *     int val;
 *     ListNode *next;
 *     ListNode(int val) {
 *        this->val = val;
 *        this->next = NULL;
 *     }
 * }
 */
class Solution { public: /** * @param head: n * @return: The new head of reversed linked list. */ ListNode * reverse(ListNode * head) { // write your code here if(head == NULL) //連結串列為空 { return NULL; } if (head->next ==
NULL) //只有一個元素的情況 { return head; } ListNode *p = head->next; head->next = NULL; ListNode *q = p->next; while(p) { p->next = head; head = p; p = q; if(q != NULL) {
q = q->next; } } return head; } };

參考答案

/**
* 本參考程式來自九章演算法,由 @九章演算法 提供。版權所有,轉發請註明出處。
* - 九章演算法致力於幫助更多中國人找到好的工作,教師團隊均來自矽谷和國內的一線大公司在職工程師。
* - 現有的面試培訓課程包括:九章演算法班,系統設計班,演算法強化班,Java入門與基礎演算法班,Android 專案實戰班,
* - Big Data 專案實戰班,演算法面試高頻題班, 動態規劃專題班
* - 更多詳情請見官方網站:http://www.jiuzhang.com/?source=code
*/ 

/**
 * Definition of ListNode
 * 
 * class ListNode {
 * public:
 *     int val;
 *     ListNode *next;
 * 
 *     ListNode(int val) {
 *         this->val = val;
 *         this->next = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param head: The first node of linked list.
     * @return: The new head of reversed linked list.
     */
    ListNode *reverse(ListNode *head) {
        ListNode *prev = NULL;
        while (head != NULL) {
            ListNode *temp = head->next;
            head->next = prev;
            prev = head;
            head = temp;
        }
        return prev;
    }
};