19.Remove Nth Node From End of List

大三准备面试,开始做leetCode,遇到有意思的题在这里做个总结。

这是一道easy题,但我已经很久没用过链表所以还是做个总结。

做leetCode的题虽然不用自己写main函数,但我还是习惯自己写一个main函数去做测试。在写main函数的过程中发现自己已经不会建表…
弄了半天总算把建表的逻辑弄明白,建表逻辑如下:

    #define NUM 1

    struct ListNode {
        int val;
        ListNode *next;
        ListNode(int x) : val(x), next(NULL) {}
    };

    int main(){
        ListNode *root,*node;
        root = new ListNode(-1);
        root->next = NULL;
        node = root;
        for(int i = 0; i< NUM;i++){
            ListNode *now = new ListNode(i);
            node->next = now;
            node = now;
        }
        node -> next = NULL; 
    }

一开始有根节点和一个普通节点,让根节点和普通节点为统一节点以便后面的循环。这时候链表只有一个节点——根节点。然后在循环中建立一个新的节点,让原有链表的末端指向新的节点,再把链表的末端扩展到包含新节点的位置。

遍历链表:

node = root;
while(node != NULL){
    cout<<node->val<<endl;
    node = node->next;
}

节点不为空就继续往下,直到为空

现在我们再来看原问题,问题要求我们去除链表的倒数第n个节点。我这里使用的方法是用两个指针一前一后去做遍历,前面的指针要比后面的指针快n个位置,这样当前面的指针到尾的时候,后面那个指针就让当前节点的下一节点为下下节点就可以了。

ListNode* removeNthFromEnd(ListNode* head, int n) {
    ListNode *node,*node2;
    node = node2 = head;
    int temp = 0;
    while(node->next!= NULL){
        node = node->next;
        if(temp >= n){
            node2 = node2->next;
        }
        temp++;
    }
    if(temp+1==n && node2->next != NULL)
        return node2->next;
    if(node2->next == NULL){
        return NULL;
    }else if (node2->next->next == NULL && node2->next != NULL){
        node2->next = NULL;
    }else if(node2->next != NULL && node2->next->next != NULL){
        node2->next = node2->next->next;
    }
    return head;
}

首先我们要注意到:

if(node2->next == NULL){
        return NULL;
    }else if (node2->next->next == NULL && node2->next != NULL){
        node2->next = NULL;
    }else if(node2->next != NULL && node2->next->next != NULL){
        node2->next = node2->next->next;
    }

在每次使用指针的时候,我们都要注意越界的问题,必须先要进行判断该指针是否为空,如若不然很容易造成运行出错。比如说链表为1,且我要去除倒数第一个节点时,如果不判断node2->next是否为空,那么势必会因为1后面是空而造成运行出错。

同时还有当链表的长度和n是一样的时候的特殊处理:

if(temp+1==n && node2->next != NULL)
        return node2->next;

因为前面的逻辑只能去掉node2之后的节点或者直接返回空,如果遇到1->2,n=2时就会答案错误。

以上代码是我通过leetCode的代码,但在写这篇博客的时候,我突然意识到还是有问题,如果传进来的根节点就是空的,那我的代码还是会运行出错,所以应该再加一个根节点是否为空的判断。

总之遇到指针的问题必须要时时刻刻注意指针越界的问题。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值