给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
思路:利用定义两个都指向首节点的指针,第一个指针先移动n次,然后第二个指针移动与第一个指针一起移动,当第一个指针移动到末尾指针的时候,第二个指针刚好移动到倒数第n+1个节点,修改指针指向即可。需要注意临界情况
/**
* 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) {
if(head==nullptr || head->next==nullptr)
return nullptr;
ListNode* first = head;
ListNode* second = head;
for(int i=0;i<n;++i){
first = first->next;
if(first->next==nullptr && i<n-1)
return second->next;
}
while(first->next!=nullptr){
first = first->next;
second = second->next;
}
if(second->next)
second->next = second->next->next;
return head;
}
};