给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
思路:
参考了网上的方法,采用了双指针,一个指针先向后移动n位,第二个指针这时候开始移动。
始终保持两个指针距离为n,当第一个指针到达末尾时,第二个指针恰好指向要删除节点
为了方便删除,当第一个指针到达末尾节点时,停止移动。
此时第二个指针在要删除节点前一个节点位置。
具体代码
/**
* 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) {
int count;
ListNode* first = head;
ListNode* second = head;
//两指针中间包含n-1个节点,这样second才指向删除节点前一个
for(int i = 0; i < n; i++){
if(first == NULL){
return NULL;
}
first = first->next;
}
//如果要删除的是头节点,那么直接返回第二个节点即可
if(first == NULL){
return head->next;
}
//继续移动,保持距离
while(first->next != NULL){
first = first->next;
second = second->next;
}
//删除节点
second->next = second->next->next;
return head;
}
};
网友评论