数据结构习题--删除链表倒数的第N个结点

数据结构习题–删除链表倒数的第N个结点

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

方法：双指针(推荐)

分析

因为该链表是单链表，我们很难实现往回遍历，所以找到最后一个结点，然后再往回一个一个遍历是不现实的，但是我们可以通过双指针实现类似的效果，如果要删除倒数第N个节点，我们可以让快指针比慢指针先走N个结点，当快指针达到链表的，末尾时，此时慢指针的结点就是我们题中需要的结点

注意：以下几点需要考虑

1. 我们要找到是删除掉的结点的前一个结点
2. 如果要删除的结点是头结点，那么其没有前面一个结点，需要单独处理，并更新头结点

代码

package LinkList;

public class deleteNodeFromEnd {

    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) {
            int count = 0;
            ListNode fast = head;
            ListNode slow = head;
            // 最后循环结束，快指针在链表末尾的下一个null处，相当于遍历了 N 次(N 为链表长度)
            // 我们需要删除掉倒数第n个，那么其就是第 N - n + 1个结点
            // 而进行删除操作，我们需要找到它的前驱节点，那么就应该找到 第N - n个结点
            // 那么快慢指针之间就应该间隔 N 减去 N - n 个位置，即 n 个位置
            // 因为初始的遍历位置就是一个结点，所以需要间隔再加一个位置，即 n + 1
            while (fast != null){
                fast = fast.next;
                // 判断间隔位置，是否需要移动慢指针
                if (count > n){
                    slow = slow.next;
                }
                count++;
            }
            // 因为初始为置为第一个结点，后面需要遍历 N - 1个结点，最后达到末尾结点，还有一次取更新到null
            // 所以总共遍历了N次，即count加到N，如果要删除第一个结点，即倒数第 N 个结点
            // 那么此时只需要判断 count 是否等于 n，就知道是不是要删除掉是头结点，从而完成特殊处理
            if (count == n){
                // 删除头结点时，特殊操作，需要更新头结点
                head = head.next;
            }else {
                // 正常的删除流程
                slow.next = slow.next.next;
            }
            return head;
        }

复杂度

时间复杂度O（n）
空间复杂度O（1）

方法：栈

分析

我们也可以在遍历链表的同时将所有节点依次入栈。根据栈「先进后出」的原则，我们弹出栈的第 n 个节点就是需要删除的节点，并且目前栈顶的节点就是待删除节点的前驱节点。这样一来，删除操作就变得十分方便了。

代码

class Solution {
    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode dummy = new ListNode(0, head);
        Deque<ListNode> stack = new LinkedList<ListNode>();
        ListNode cur = dummy;
        while (cur != null) {
            stack.push(cur);
            cur = cur.next;
        }
        for (int i = 0; i < n; ++i) {
            stack.pop();
        }
        ListNode prev = stack.peek();
        prev.next = prev.next.next;
        ListNode ans = dummy.next;
        return ans;
    }
}

复杂度

时间复杂度O(n)
空间复杂度O(n)