題目:輸入一個鏈表的頭結(jié)點,反轉(zhuǎn)該鏈表,并返回反轉(zhuǎn)后鏈表的頭結(jié)點。鏈表結(jié)點定義如下:
struct ListNode
{
int m_nKey;
ListNode* m_pNext;
};
分析:這是一道廣為流傳的微軟面試題。由于這道題能夠很好的反應(yīng)出程序員思維是否嚴(yán)密,在微軟之后已經(jīng)有很多公司在面試時采用了這道題。
為了正確地反轉(zhuǎn)一個鏈表,需要調(diào)整指針的指向。與指針操作相關(guān)代碼總是容易出錯的,因此在動手寫程序之前作全面的分析。在面試的時候不急于動手而是一開始做仔細(xì)的分析和設(shè)計,將會給面試官留下很好的印象,因為在實際的軟件開發(fā)中,設(shè)計的時間總是比寫代碼的時間長。與其很快地寫出一段漏洞百出的代碼,遠(yuǎn)不如用較多的時間寫出一段健壯的代碼。
為了將調(diào)整指針這個復(fù)雜的過程分析清楚,我們可以借助圖形來直觀地分析。假設(shè)下圖中l(wèi)、m和n是三個相鄰的結(jié)點:
a戀…氀 mànà…
假設(shè)經(jīng)過若干操作,我們已經(jīng)把結(jié)點l之前的指針調(diào)整完畢,這些結(jié)點的m_pNext指針都指向前面一個結(jié)點,F(xiàn)在我們遍歷到結(jié)點m。當(dāng)然,我們需要把調(diào)整結(jié)點的m_pNext指針讓它指向結(jié)點l。但注意一旦調(diào)整了指針的指向,鏈表就斷開了,如下圖所示:
a戀…l洀 nà…
因為已經(jīng)沒有指針指向結(jié)點n,我們沒有辦法再遍歷到結(jié)點n了。因此為了避免鏈表斷開,我們需要在調(diào)整m的m_pNext之前要把n保存下來。
接下來我們試著找到反轉(zhuǎn)后鏈表的頭結(jié)點。不難分析出反轉(zhuǎn)后鏈表的頭結(jié)點是原始鏈表的尾位結(jié)點。什么結(jié)點是尾結(jié)點?就是m_pNext為空指針的結(jié)點。
基于上述分析,我們不難寫出如下代碼:
///////////////////////////////////////////////////////////////////////
// Reverse a list iteratively
// Input: pHead - the head of the original list
// Output: the head of the reversed head
///////////////////////////////////////////////////////////////////////
ListNode* ReverseIteratively(ListNode* pHead)
{
ListNode* pReversedHead = NULL;
ListNode* pNode = pHead;
ListNode* pPrev = NULL;
while(pNode != NULL)
{
// get the next node, and save it at pNext
ListNode* pNext = pNode->m_pNext;
// if the next node is null, the currect is the end of original
// list, and it's the head of the reversed list
if(pNext == NULL)
pReversedHead = pNode;
// reverse the linkage between nodes
pNode->m_pNext = pPrev;
// move forward on the the list
pPrev = pNode;
pNode = pNext;
}
return pReversedHead;
}
擴展:本題也可以遞歸實現(xiàn)。
左旋轉(zhuǎn)字符串
題目:定義字符串的左旋轉(zhuǎn)操作:把字符串前面的若干個字符移動到字符串的尾部。如把字符串a(chǎn)bcdef左旋轉(zhuǎn)2位得到字符串cdefab。請實現(xiàn)字符串左旋轉(zhuǎn)的函數(shù)。要求時間對長度為n的字符串操作的復(fù)雜度為O(n),輔助內(nèi)存為O(1)。
分析:如果不考慮時間和空間復(fù)雜度的限制,最簡單的方法莫過于把這道題看成是把字符串分成前后兩部分,通過旋轉(zhuǎn)操作把這兩個部分交換位置。于是我們可以新開辟一塊長度為n+1的輔助空間,把原字符串后半部分拷貝到新空間的前半部分,在把原字符串的前半部分拷貝到新空間的后半部分。不難看出,這種思路的時間復(fù)雜度是O(n),需要的輔助空間也是O(n)。
接下來的一種思路可能要稍微麻煩一點。我們假設(shè)把字符串左旋轉(zhuǎn)m位。于是我們先把第0個字符保存起來,把第m個字符放到第0個的位置,在把第2m個字符放到第m個的位置…依次類推,一直移動到最后一個可以移動字符,最后在把原來的第0個字符放到剛才移動的位置上。接著把第1個字符保存起來,把第m+1個元素移動到第1個位置…重復(fù)前面處理第0個字符的步驟,直到處理完前面的m個字符。
該思路還是比較容易理解,但當(dāng)字符串的長度n不是m的整數(shù)倍的時候,寫程序會有些麻煩,感興趣的朋友可以自己試一下。由于下面還要介紹更好的方法,這種思路的代碼我就不提供了。
我們還是把字符串看成有兩段組成的,記位XY。左旋轉(zhuǎn)相當(dāng)于要把字符串XY變成YX。我們先在字符串上定義一種翻轉(zhuǎn)的操作,就是翻轉(zhuǎn)字符串中字符的先后順序。把X翻轉(zhuǎn)后記為XT。顯然有(XT)T=X。
我們首先對X和Y兩段分別進行翻轉(zhuǎn)操作,這樣就能得到XTYT。接著再對XTYT進行翻轉(zhuǎn)操作,得到(XTYT)T=(YT)T(XT)T=YX。正好是我們期待的結(jié)果。
分析到這里我們再回到原來的題目。我們要做的僅僅是把字符串分成兩段,第一段為前面m個字符,其余的字符分到第二段。再定義一個翻轉(zhuǎn)字符串的函數(shù),按照前面的步驟翻轉(zhuǎn)三次就行了。時間復(fù)雜度和空間復(fù)雜度都合乎要求。
參考代碼如下:
#include "string.h"
///////////////////////////////////////////////////////////////////////
// Move the first n chars in a string to its end
///////////////////////////////////////////////////////////////////////
char* LeftRotateString(char* pStr, unsigned int n)
{
if(pStr != NULL)
{
int nLength = static_cast(strlen(pStr));
if(nLength > 0 || n == 0 || n > nLength)
{
char* pFirstStart = pStr;
char* pFirstEnd = pStr + n - 1;
char* pSecondStart = pStr + n;
char* pSecondEnd = pStr + nLength - 1;
// reverse the first part of the string
ReverseString(pFirstStart, pFirstEnd);
// reverse the second part of the strint
ReverseString(pSecondStart, pSecondEnd);
// reverse the whole string
ReverseString(pFirstStart, pSecondEnd);
}
}
return pStr;
}
///////////////////////////////////////////////////////////////////////
// Reverse the string between pStart and pEnd
///////////////////////////////////////////////////////////////////////
void ReverseString(char* pStart, char* pEnd)
{
if(pStart == NULL || pEnd == NULL)
{
while(pStart <= pEnd)
{
char temp = *pStart;
*pStart = *pEnd;
*pEnd = temp;
pStart ++;
pEnd --;
}
}
}