劍指offer(14)連結串列中倒數第k個結點
阿新 • • 發佈:2018-12-11
題目描述
輸入一個連結串列,輸出該連結串列中倒數第k個結點。
解題思路
輸出的是 倒數第k個連結串列,可以參考Leetcode刪除連結串列倒數節點的那個演算法題
一共有兩種解法,一是兩遍遍歷的,二是一遍遍歷的。
兩次遍歷的思路就是先遍歷一遍 求出連結串列長度,然後在遍歷,輸出N-k+1的節點。這道題靠魯棒性,需要考慮各種情況,比如可能不存在倒數第k個結點。
程式碼如下
/* struct ListNode { int val; struct ListNode *next; ListNode(int x) : val(x), next(NULL) { } };*/ class Solution { public: ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) { ListNode *res=new ListNode(0); ListNode *first=pListHead; int length=0; while(first!=NULL) { first=first->next; length++; } length-=k; if(length<0) return 0; first=pListHead; while(length>0) { length--; first=first->next; } res=first; return res; } };
一次便利的結果,就是宣告兩個指標,兩個指標都指向頭結點,第一個指標向後移動k+1位,然後,兩個指標同步向後執行,這個時候,兩個指標之間的距離就是k,直至投一個指標挪到最後一位,此時,輸出第二個指標的結點即可。
程式碼如下
/* struct ListNode { int val; struct ListNode *next; ListNode(int x) : val(x), next(NULL) { } };*/ class Solution { public: ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) { ListNode *res=new ListNode(0); ListNode *first=pListHead; ListNode *second=pListHead; while(k>0) { if(first!=NULL) first=first->next; else return 0; k--; } while(first!=NULL) { first=first->next; second=second->next; } res=second; return second; } };