LeetCode Remove Nth Node From End of List

来源:互联网 发布:小米系统内存优化级别 编辑:程序博客网 时间:2024/05/17 18:17
Q:

Given a linked list, remove the nth node from the end of list and return its head.

For example,

   Given linked list: 1->2->3->4->5, and n = 2.   After removing the second node from the end, the linked list becomes 1->2->3->5.

Note:
Given n will always be valid.
Try to do this in one pass.

        题目大意是删除链表的倒数第n个结点,这里从1开始计数,既然要求的是删除,那么我们最好找到的应该是倒数第n+1个结点,比如说上题n=2,我们应该让指针指向倒数第三个结点即结点3,这样就可以通过3->next = 3->next->next了,但是这里要求的是最多只能一次遍历,所以说先遍历一次求出链表节点个数k 然后通过指针遍历k-n个结点来确定倒数第n+1个结点是不行的。为了实现只遍历链表一次就能找到倒数第n+1个结点,我们可以定义两个指针。第一个指针从链表的头指针开始遍历向前走n步, 第二个指针保持不动;从第n+1步开始,第二个指针也开始从头指针开始遍历。由于两个指针的距离保持在n,当第一个指针到达链表的尾结点时,第二个指针指向的刚好就是倒数第n+1个结点。


盗的图>_< 这是在有6个结点的链表上删除倒数第2个结点的过程。第一个指针先走2步,第二个指针随后开始走,p1到链表尾部时,p2就到了倒数第n+1个结点那儿了。

        这题值得注意的是他有可能删除的是头结点,比如只有一个结点的链表让你删除倒数第1个结点。一般需要改变头结点的,我们都可以创建一个辅助的头指针,这样是为了修改链表头的方便。既然多了一个结点,那么我们的n相应也要+1。具体看代码。

代码:

class Solution {public:    ListNode *removeNthFromEnd(ListNode *head, int n) {if(head == NULL || n == 0)return NULL;ListNode * helper = new ListNode(0);helper->next = head;ListNode *p1 = helper, *p2 = helper;n++;//因为多加了个节点所以n也要+1while(n > 0 && p1){p1 = p1->next;n--;}while(p1){p1 = p1->next;p2 = p2->next;}p2->next = p2->next->next;return helper->next;    }};


0 0
原创粉丝点击