给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
示例 2:
输入:head = [1], n = 1
输出:[]
示例 3:
输入:head = [1,2], n = 1
输出:[1]
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
思路:双指针法:
1.定义两个指针p,q分别指向头节点,令q先向后走n个节点,之后p与q同时向后遍历,当q->next为NULL时,p即为要删除节点的前驱,令p->next=p->next->next即可。
2.特殊情况,当删除第一个节点时,即倒是第n个 n=n时,此时q向后走n个节点之后为NULL,不存在q->next,此时直接特判处理head=head->next即可。
AC代码:(C++)
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode *p = head, *q = head;
for (int i = 0; i < n; i++) {
q = q->next;
}
if (q != NULL) { //如果不是删除第一个节点
while (q->next != NULL) {
q = q->next;
p = p->next;
}
p->next = p->next->next;
} else { //删除第一个节点
head = head->next;
}
return head;
}
};