给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5. 说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:这里可以用快慢指针,先找到倒数第k个节点(找到倒数第k个节点参考面试题 02.02. 返回倒数第 k 个节点),然后直接删除即可,但是这里因为牵涉到链表的删除,所以我们需要保存一下前驱节点的指针,并且删除头节点的情况需要单独考虑,代码如下:
/** * 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) { ListNode* pre = NULL; ListNode* p1 = head; ListNode* p2 = head; for(int i = 0;i < n;i++) p1 = p1->next; while(p1) { p1 = p1->next; pre = p2;//保存以前节点 p2 = p2->next; } if(p2 == head)//如果p2恰好是头节点 { return head->next; } else { pre->next = p2->next; return head; } } };