给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2. 当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
思路:双指针,控制两个指针距离为n,同时后移,然后删除第一个指针的下一个元素,注意不要访问空指针,比如p->next,p->next->next都不能为空。
我的解答:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
static const auto _ = []() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
return nullptr;
}();
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
if(!(head->next))return NULL;
ListNode* p=head, * q=head;
int i=0;
while(i<n&&q->next)
{
q=q->next;
i++;
}
if(i<n)
return head->next;
while(q->next)
{
p=p->next;
q=q->next;
}
if(p->next->next)
p->next=p->next->next;
else
p->next=NULL;
return head;
}
};