5. 链表 - 删除链表的倒数第N个节点

1.题目

LeetCode: 19. 删除链表的倒数第N个节点

【medium】

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

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.

当删除了倒数第二个节点后，链表变为 1->2->3->5.

说明：

给定的 n 保证是有效的。

进阶：

你能尝试使用一趟扫描实现吗？

2.解题

方法一：一次遍历

遍历链表，每次遍历一个节点执行一次n–，如果n<=0，则用另一个指针进行记录要删除节点的前一个节点，最后执行q.next = q.next.next，将节点删除，返回head。

如果链表遍历结束后n不为0，则说明需要删除的元素是原链表的第一个元素，所以直接返回head.next即可。

java：

/**
 * 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 p = head, q = head;
        while (p.next != null) {
            if (n <= 0) {
                q = q.next;
            }
            p = p.next;
            n--;
        }
        if (n == 1) return head.next;   // 如果要删除的节点为第一个节点，则直接返回head.next
        q.next = q.next.next;
        return head;
    }
}

时间复杂度：O(n)，由于只遍历了一次，所以时间复杂度取决于链表长度

空间复杂度：O(1)

方法二：栈

利用栈先进后出的特点，遍历链表，将所有节点放入栈中，然后出栈n次，将第n个节点删除。

java：

/**
 * 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) {
        Deque<ListNode> stack = new LinkedList<ListNode>();
        ListNode p = head;
        int num = 0;
        while (p != null) {
            stack.push(p);
            num++;
            p = p.next;
        }
        if (num == n) return head.next;
        while (n-- > 0) {
            stack.pop();
        }
        ListNode pre = stack.peek();
        pre.next = pre.next.next;
        return head;
    }
}

时间复杂度：O(n)，时间复杂度取决于链表长度

空间复杂度：O(n)，由于用到了额外的存储空间，栈