leetcode删除链表的倒数第 n 个结点
思想其实很简单:设置一个tail指针,再设置一个距离tail为n步的指针p,当tail指向尾部时,指针p就是倒数第n个节点,所以只需要一遍遍历指针。
为什么要伪头节点?
我这里记录的是prep不是p,方便删除p节点,伪头节点方便处理边界情况,假设head=[1],n=1的特殊情况,要引入1个头节点合适
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* prehead=new ListNode();
prehead->next = head;
ListNode* tail = prehead, * prep = prehead;
while (n-- > 0) {
tail = tail->next;
}
while (tail->next != nullptr) {
tail = tail->next;
prep = prep->next;
}
prep->next = prep->next->next;
return prehead->next;
}