Medium
Remove Nth Node From End of List — C++
Full explanation · Time O(n) · Space O(1)
// Time: O(n)
// Space: O(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) {
ListNode dummy{0};
dummy.next = head;
auto slow = &dummy;
auto fast = &dummy;
// fast is n-step ahead.
while (n > 0) {
fast = fast->next;
--n;
}
// When fast reaches the end, slow must be nth to last node.
while (fast->next != nullptr) {
slow = slow->next;
fast = fast->next;
}
auto node_to_delete = slow->next;
slow->next = slow->next->next;
delete node_to_delete;
return dummy.next;
}
};