19.Remove Nth Node From End of List

来源:互联网 发布:android开发特效源码 编辑:程序博客网 时间:2024/06/15 21:52

大三准备面试,开始做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的代码,但在写这篇博客的时候,我突然意识到还是有问题,如果传进来的根节点就是空的,那我的代码还是会运行出错,所以应该再加一个根节点是否为空的判断。

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

2 0
原创粉丝点击