LeetCode 連結串列專題 19 刪除倒數第n個數 java 遞迴解法 幾乎雙百 思路簡單

題目詳情

本題是刪除連結串列的倒數第n個數，因為連結串列的性質，所以必須得遍歷才知道連結串列有多長。
為了知道每個節點是倒數第幾個節點，我們這裡用回溯來計數，從而統計自己是第幾個數。
另外我們刪除第n個數的邏輯是，用第n+1個數去指向n-1個數（倒數）。
不過這存在一個特殊情況。n-1的數必然存在，它可以為空，但n+1的數可能不存在，當我們要刪除的數就在連結串列的頭部時，n+1這個元素為空，我們可以用一個節點指向空，但絕不能用空節點指向另外一個節點。
所以我們需要去判斷第n個元素是否是頭節點，如果是，那麼我們就返回頭結點的next即可。
程式碼的效果是100%和99.19%

程式碼

class Solution {
    private boolean flag = false;// 用以強制退出遞迴
    ListNode nSubOne = null,nPlusOne = null,nn=null;
    public ListNode removeNthFromEnd(ListNode head, int n) {
        backtrack(head,n);
        //只需要考慮一個特殊情況：刪除的是頭元素，則無法用n+1->n-1來刪除n，那麼直接返回head的next即可
        if(nn==head)
            return head.next;
        nPlusOne.next = nSubOne;
        return head;
    }
    public int backtrack(ListNode p, int n){
        if(flag)
            return 0;
        //判斷是否為空
        if(p==null)
            return 0;
        int count = backtrack(p.next,n);
        count++;
        //進行判斷，找到倒數n-1,n和n+1個數
        if(count==n-1)
            nSubOne = p;
        else if(count==n+1){
            nPlusOne = p;
            flag = true;
        }
        else if(count==n)
           nn = p;
        return count;
    }
}