leetcode 题号#19 删除链表的第n个节点

查看题目详情可点击此处。

题目

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

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后，链表变为 1->2->3->5.

说明：
给定的 n 保证是有效的。

解题思路

题目中需要删除倒数第个结点，关键在于找到倒数第个结点，其实换句话说也就是从此结点数到尾结点总共有个结点，最直观的方法就是遍历链表结点，从此结点往后遍历记数到第个时判断是否为尾结点，如果是将此结点删除即可。

分析一下复杂度，因为没有申请额外的空间，所以空间复杂度为，假设链表长度为m，需要删除倒数第n个结点，遍历到当前结点需要遍历链表次，每个结点又需要遍历到之后的第个结点判断是否为尾结点，这种方法需要使用嵌套循环，所以时间复杂度为。

转换一下思路，其实可以先使用头结点后的第个结点，如果此结点（以下称为结点A）是尾结点那头结点就是倒数第个结点，如果结点A不是尾结点，可以从结点A开始遍历，同时倒数第个结点为头结点的下一结点，循环遍历，直到找到尾结点。

这种方法的需要两个循环，假设链表长度为m，需要删除倒数第n个结点，获得头结点后的第个结点需要遍历次，而第二个循环为了查找到尾结点需要循环次，所以时间复杂度为，最差情况下时间复杂度为，而之前一种方法 只有在最好情况下时间复杂度才为，两个方法同样空间复杂度都为，所以第二种方法会更好，代码如下。

class Solution {
    public ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode tailNode = moveToNthNode(head, n);

        ListNode currNode = head;
        ListNode prevNode = null;
        while(tailNode != null && tailNode.next != null) {
            prevNode = currNode;
            currNode = currNode.next;
            tailNode = tailNode.next;
        }
        return deleteNode(head, prevNode, currNode);
    }
    
    private ListNode moveToNthNode(ListNode head, int n) {
        int i = 1;
        ListNode currNode = head;
        while(i < n) {
            currNode = currNode.next;
            ++i;
        }

        return currNode;
    }
    
    private ListNode deleteNode(ListNode head, ListNode prevNode, ListNode removeNode) {
        ListNode headNode = head;
        if(prevNode == null) {
            headNode = removeNode.next;
        } else {
            prevNode.next = removeNode.next;
        }
        return headNode;
    }
}