LeetCode19. Remove Nth Node From End of List

一、问题描述

Given a linked list, remove the nth node from the end of list and return its head.

For example,

Given linked list: 1->2->3->4->5, and n = 2.

After removing the second node from the end, the linked list
becomes 1->2->3->5.

Note:
Given n will always be valid.
Try to do this in one pass.

二、解题思路

一般而言，要想删除倒数第n个结点，只需要先遍历一次链表得到链表的长度L，再遍历第二次找到就可以找到要删除的结点，但是这种方法需要遍历两次链表。另一个方法是，使用两个指针变量p、q，q先行，之后同时移动p、q，保持p、q之间的距离为n+1个结点，当q指向链表尾的空结点时，p刚好指向要被删除结点的上一个结点。考虑到链表长度为1时的特殊情况，这里需要借助一个新结点beginNode，它的下一个结点即为head结点。

三、C++代码

Definition for singly-linked list.
struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode* beginNode = new ListNode(0);
        beginNode->next = head;
        ListNode* q = head;
        ListNode* p = beginNode;
        while (n--) {
            q = q->next;
        }
        while (q != NULL) {
            q = q->next;
            p = p->next;
        }
        ListNode* temp = p->next;
        p->next = temp->next;
        delete temp;
        return beginNode->next;
    }
};

四、遇到的问题

一开始由于对指针知识有点遗忘导致难以发现的指针错误。在链表长度为1的情况下，将temp指针delete的同时也delete了head指针，而函数最后又返回head指针，导致错误，将return head修改为return beginNode->next就解决了问题。