Lintcode 删除链表中倒数第k个结点

来源:互联网 发布:拓扑图算法 编辑:程序博客网 时间:2024/05/21 10:50

给定一个链表,删除链表中倒数第n个节点,返回链表的头节点。

样例
给出链表1->2->3->4->5->null和 n = 2.

删除倒数第二个节点之后,这个链表将变成1->2->3->5->null.

挑战
O(n)时间复杂度

要考虑的测试用例情况(len表示链表长度):
1. n>len,
2. 要删除的结点是头节点或只有一个结点(只有一个结点要特殊处理,因为没有前驱也没有后继)
3. 要删除的结点是最后一个结点(要从头开始找到最后一个结点前的一个结点)
4. 要删除的结点在链表中间(不是最后一个也不是第一个)
我的想法:
首先遍历一遍链表,计算链表的长度(for语句停止条件注意),然后找到倒数第n个待删除结点,倒数第n个结点和第n+1个结点值交换一下,删除倒数第n+1个结点。如果n=1,则需重新遍历一遍,找到前驱,特殊处理。复杂度O(n),但是基本上要遍历2次。

/** * Definition of ListNode * class ListNode { * public: *     int val; *     ListNode *next; *     ListNode(int val) { *         this->val = val; *         this->next = NULL; *     } * } */class Solution {public:    /**     * @param head: The first node of linked list.     * @param n: An integer.     * @return: The head of linked list.     */    ListNode *removeNthFromEnd(ListNode *head, int n) {        // write your code here        if(head == NULL){            return NULL;        }        if(n<=0)        {            return nullptr;        }        int len = 0;        ListNode *p = head;        for(p; p!=NULL; p= p->next){            len++;        }        if(len == 1&& n == 1){            return NULL;        }        p = head;        for(int i = 1; i<=len-n; i++){            p = p->next;        }//p at nth node        if(p->next == NULL){            ListNode *pre;            for(pre = head;pre->next!=p; pre = pre->next);            pre->next = NULL;            return head;        }       ListNode *q = p->next;       int tmp = q->val;       q->val = p->val;       p->val = tmp;       p->next = q->next;       return head;    }};

然而这题实际上有更好的解法:双指针,一个指针先走n-1步

/** * Definition of ListNode * class ListNode { * public: *     int val; *     ListNode *next; *     ListNode(int val) { *         this->val = val; *         this->next = NULL; *     } * } */class Solution {public:    /**     * @param head: The first node of linked list.     * @param n: An integer.     * @return: The head of linked list.     */    ListNode *removeNthFromEnd(ListNode *head, int n) {        // write your code here        if(head == NULL||n<=0){            return NULL;        }        ListNode *pHead = head;        for(int i = 0; i< n-1; i++){ //中途可能会断掉,len<n的情况            pHead = pHead ->next;            if(pHead == NULL){//提前返回                return NULL;            }        }        ListNode *q = head;        if(q->next == NULL){//只有一个结点,直接返回            return NULL;        }        while(pHead->next!=NULL){            pHead = pHead->next;            q = q->next;        }        if(q->next == NULL){//最后一个结点,再遍历            ListNode *pre = head;            while(pre->next != q){                pre = pre->next;            }            pre->next = NULL;            return head;        }else{            int tmp = q->next->val;            q->next->val = q->val;            q->val = tmp;            q->next = q->next->next;            return head;        }    }};
0 0
原创粉丝点击