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

给定一个链表,删除链表中倒数第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;
        }


    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值