給定一個(gè)鏈表股耽,刪除鏈表的倒數(shù)第 n 個(gè)節(jié)點(diǎn),并且返回鏈表的頭結(jié)點(diǎn)钳幅。
示例:
給定一個(gè)鏈表: 1->2->3->4->5, 和 n = 2.
當(dāng)刪除了倒數(shù)第二個(gè)節(jié)點(diǎn)后物蝙,鏈表變?yōu)?1->2->3->5.
說明:
給定的 n 保證是有效的。
刪除倒數(shù)第n個(gè)節(jié)點(diǎn)敢艰,用兩個(gè)指針诬乞,先讓后一個(gè)便利n個(gè)節(jié)點(diǎn),然后兩個(gè)指針一起往后走钠导,當(dāng)后頭的走到最后震嫉,前頭的下一個(gè)節(jié)點(diǎn)就是要?jiǎng)h除的節(jié)點(diǎn),然后把節(jié)點(diǎn)刪掉牡属,最后把原來鏈表head返回票堵。
有一個(gè)特殊情況,就是要?jiǎng)h除的節(jié)點(diǎn)是第一個(gè)節(jié)點(diǎn)逮栅,刪掉之后head就空了悴势。所以可以添加一個(gè)頭節(jié)點(diǎn)窗宇,就不會(huì)出現(xiàn)這種特殊情況了, 結(jié)果返回頭節(jié)點(diǎn)的next特纤。
代碼如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* headnode = new ListNode(0);
headnode->next = head;
ListNode* q = headnode;
ListNode* p = headnode;
for(int i = 0; i < n; i++)
q = q->next;
while(q->next != NULL)
{
q = q->next;
p = p->next;
}
q = p->next;
p->next = p->next->next;
delete(q);
ListNode* r = headnode->next;
delete(headnode);
return r;
}
};