leetcode19. 删除链表的倒数第 N 个结点 | 思路+解题过程 | 简单解法 | 优化解法

题目

给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。

简单思路 —— 两次遍历

超绝简单思路 ：第一个指针遍历第一遍链表：计算结点总数 ；第二个指针遍历第二遍链表：删除结点 (删除头结点的情况单独判断)

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode node = head;
        ListNode node2 = head;
        
        //第一个指针遍历链表，计算结点总数
        int cnt = 0;
        while(node != null){
            cnt = cnt + 1;
            node = node.next;
        }
        
        //如果要删除的是第一个结点
        if(n == cnt){ 
            return head.next;
        }
        
        //如果删除的不是第一个结点
        //第二个指针遍历一直到待删除的结点前，做删除工作
        for (int i = 0; i < cnt-n-1; i++) {
            node2 = node2.next;
        }
        node2.next = node2.next.next;
        return head;
    }
}

优化思路 —— 一次遍历

两个指针，一次遍历

大致思路：

假设链表的结点总数为n，想要删除倒数第k个结点(n未知，k已知)。逆序遍历就遍历k次到达该结点，顺序遍历就遍历n-k次到达该结点。

问题是如何才能让指针顺序遍历n-k步呢？可以让第一个指针先遍历k步，那么剩下n-k个结点，此时让第二个指针从头开始和第一个指针一起遍历，当第一个指针到终点时，第二个指针就走到了要删除的那个结点！

解题过程：

---- 指针B(第二个指针)初始位置指向头结点(第一个结点)，若想走到倒数第k个结点，就从头结点开始走n-k步，即可指向该待删除结点。
(可举例验证)

---

---- 但要删除该结点，需要指向该结点的前一个结点，即指针B需要从头结点开始走n-k-1步，再用node.next=node.next.next即可. (普通情况，不考虑要删除第一个结点的情况)

---- 指针A(第一个指针)初始位置指向头结点，A走了k步后，已经遍历了k+1个结点，若想走到最后一个结点停止，还需要走n-k-1步.

---

---- 所以，当指针A从头结点开始走了k步后，此时让指针B指向头结点和指针A一起往后遍历，当指针A走到最后一个结点的时候，指针B就走到了待删除结点的前一个结点！！

---

---- 若要删除的结点是头结点，单独判断一下即可

时间复杂度:
O(N)

Code

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode removeNthFromEnd(ListNode head, int k) {
            ListNode node1 = head;
            //第一个指针走k步
            for (int i = 0; i < k; i++) {
                node1 = node1.next;
            }
            //此时让第二个指针指向头结点，准备和第一个指针一起遍历
            ListNode node2 = head;
            //当第一个指针遍历到最后一个结点时(node1.next!=null)，第二个指针就可以做删除工作了
            //特殊情况：当k==n时，即需要删除头结点时，特殊判断；node1遍历k次后，直接就指向了空(node1!=null)
            while(node1!=null && node1.next!=null){
                node1 = node1.next;
                node2 = node2.next;
            }
            
            if(node1 == null){
                //说明需要删除的是头结点，只需要逻辑返回头结点的下一个结点即可
                return head.next;
            } else{
                //删除的不是头结点，第二个指针做逻辑删除工作
                node2.next = node2.next.next;
                return head;
            }
    }
}