Remove Nth Node From End of List

標籤：style   blog   color   io   for   div   代碼   log   

這道題可以用雙指標的方法解，將兩個指標p,q的距離保持在n-1，然後移動q到List的最後一個元素，那麼此時p指向的便是the Nth node from the end。要刪除這個node，要分兩種情況，一種是該node為head，另一種該node是中間node。對於第一種情況，可以簡單的採用以下操作: head = head->next。對於第二種情況，我們需要引入額外的一個指標prev_p指向p的前一個node，這樣便可以通過prev_p->next = prev_p->next->next刪除該node。代碼如下：

 1 class Solution { 2 public: 3     ListNode *removeNthFromEnd(ListNode *head, int n) { 4         ListNode * p = head, * q = head, * prev_p = NULL; 5         for(int i = 0; i < n-1; i++) 6             q = q->next; 7         for(;q->next;q = q->next, prev_p = p,p = p->next); 8         if(p == head) head = head->next; 9         else 10             prev_p->next = p->next;11         return head;12     }13 };