1. 程式人生 > >leetcode刷題,總結,記錄,備忘 19

leetcode刷題,總結,記錄,備忘 19

leetcode19Remove Nth Node From End of List

Given a linked list, remove the nth node from the end of list and return its head.

For example,

   Given linked list: 1->2->3->4->5, and n = 2.

   After removing the second node from the end, the linked list becomes 1->2->3->5.

Note:


Given n will always be valid.
Try to do this in one pass.

題目的提示是使用雙指標,,,可是我最開始最先想到的是用遞迴,感覺有點非主流。。。。以前在看c和指標這本書的時候遇到一個題,將連結串列倒置,就是用的遞迴,主要思路是一路重複呼叫進去,到最後的節點處,然後一個一個返回。這題的思路就是遞迴呼叫,一個引數是連結串列節點的指標,一個引數是一個int指標,代表倒數的n個節點,先一路走到最後一個節點,然後根據n的值不斷自減,然後一直返回,如果n減到0,代表該刪除這個節點,就把返回的節點當作返回值返回上層,然後在上一層中鏈在上一層的next後面,詳細看程式碼。
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode * function(ListNode * head, int * n)
    {
        ListNode * result;
        
        if (head->next)
        {
            result = function(head->next, n);
        }
        else
        {
            result = NULL;
        }
        
        if (--*n == 0)
        {
            return result;
        }
        else
        {
            head->next = result;
            return head;
        }
    }
    
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        int m = n;
        if (head == NULL)
        {
            return NULL;
        }
        
        ListNode * result = function(head, &m);
        
        return result;
    }
};
討論區看到一個非常厲害的解決方法,用2個指標,先用一個指標走n-1個位置,然後第二個指標與第一個指標一起走,直到第一個指標為null,此時第二個指標就是要刪除的節點,並且第二個指標使用的是二級指標的方式,可以直接通過二級指標的地址修改該節點上的值,非常巧妙。
class Solution
{
public:
    ListNode* removeNthFromEnd(ListNode* head, int n)
    {
        ListNode** t1 = &head, *t2 = head;
        for(int i = 1; i < n; ++i)
        {
            t2 = t2->next;
        }
        while(t2->next != NULL)
        {
            t1 = &((*t1)->next);
            t2 = t2->next;
        }
        *t1 = (*t1)->next;
        return head;
    }
};