leetcode:Remove Nth Node From End of List

来源:互联网 发布:国外电脑网络加速器 编辑:程序博客网 时间:2024/06/06 14:12

题目

给定一个列表,删除倒数第n个节点,并且返回表头,例如:
给定列表:1->2->3->4->5, n = 2
返回:1->2->3->5

假设给定的n值都是合法的。

注意

  • 给定空表,返回NULL
  • 如果删除第一个节点,要保证不要删了表头

思路

设置两个指针p,q,相差n1,那么如果后一个q指针知道表尾时,前一个指针p所在位置即为要删除的节点,记录p的直接前驱,删除p即可。

struct ListNode {    int val;    ListNode *next;    ListNode(int x) : val(x), next(NULL) {}  };class Solution {public:    ListNode* removeNthFromEnd(ListNode* head, int n)    {        if (NULL == head) {return head;}        ListNode *p = head;        ListNode *q = head;        ListNode *pPre = NULL;//记录 p 指针直接前驱        for(int i = 0;i != n - 1;++i) q = q->next;        while(q->next)        {            pPre = p;            q = q->next;            p = p->next;        }        if(NULL == pPre)//要删除的节点是第一个节点        {            head = p->next;            delete p;        }        else        {            pPre->next = p->next;            delete p;        }        return head;    }};
0 0
原创粉丝点击