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