每日一道算法面試題（14）：leecode19 刪除鏈表的倒數第N個節點

1.算法題目

給定一個鏈表，刪除鏈表的倒數第 n 個節點，並且返回鏈表的頭結點。

示例：

給定一個鏈表: 1->2->3->4->5, 和 n = 2.

當刪除了倒數第二個節點後，鏈表變爲 1->2->3->5.

說明：

給定的 n 保證是有效的。

進階：

你能嘗試使用一趟掃描實現嗎？

通過次數131,025提交次數3

2.算法思路

雙鏈表法：可以通過遍歷一次鏈表，也不需要計算鏈表的長度，即可找到要刪除的節點，具體思路：定義兩個鏈表 firstNode 和 secondNode ，初始化時都指向頭節點，刪除鏈表的倒數第 n 個節點時需要找到倒數第 n + 1 個節點，先遍歷 firstNode，firstNode 遍歷到 n + 1 個節點的時候再開始遍歷 secondNode，當firstNode遍歷到尾結點的時候，secondNode 剛好遍歷到倒數
n + 1 個節點的位置，此時刪除 secondNode 的下一個節點即可。需要注意的是鏈表只有一個節點的特殊情況，此時需要刪除頭節點。

3.算法代碼

根據雙鏈表法的算法思路編寫的算法代碼如下：

    // 鏈表定義
    public class ListNode {
        int val;
        ListNode next;
        ListNode(int x) { val = x; }
    }

    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode firstNode = head;
        ListNode secondNode = head;
        while (firstNode != null && firstNode.next != null) {
            firstNode = firstNode.next;
            if(n == 0) {
                secondNode = secondNode.next;
            }
            if (n > 0) {
                n --;
            }
        }
        if (n > 0) { // 刪除當前節點
            if (secondNode.next == null) { // 刪除頭節點
                head = null;
            } else {  // 刪除當前節點
                head = head.next;
            }
        } else {
            ListNode tmp = secondNode.next; // 待刪除節點
            secondNode.next = tmp.next;
        }
        return head;
    }

4.總結

雙鏈表法在有關鏈表的算法中比較常用，而且用起來很方便，比如判斷一個鏈表是否有環、刪除指定位置的節點、判斷兩個鏈表是否交叉（交叉的鏈表後面的節點都是一樣的，可以讓長度長的鏈表先走兩個鏈表長度只差的位置）等等。

附判斷兩個鏈表是否交叉的算法代碼：

    // 鏈表定義
    public class ListNode {
        int val;
        ListNode next;
        ListNode(int x) { val = x; }
    }

    //獲取鏈表長度
    private int getLength(ListNode head){
        int len = 0 ;
        for(ListNode cur =head;cur!= null;cur = cur.next){
            len ++;
        }
        return len;
    }

    //判斷交叉+求交叉結點
    public ListNode getIntersectionNode(ListNode headA, ListNode headB) {
        int lenA = getLength(headA);
        int lenB = getLength(headB);

        ListNode longer = headA;
        ListNode shorter = headB;

        int diff = lenA-lenB;

        if(lenA < lenB){
            diff = lenB-lenA;
            longer = headB;
            shorter = headA;
        }

        //讓長鏈表先走diff步,保證下一步while值對比循環初始兩個鏈表長度是相同的
        for(int i = 0; i < diff; i++){
            longer =longer.next;
        }

        //循環遍歷，當值相等時停止循環
        while(longer != shorter){
            longer = longer.next;
            shorter = shorter.next;
        }
        //返回交叉結點
        return longer;
    }

如果你有疑問或更好的算法思路，歡迎留言交流！！！