leetcode 19. Remove Nth Node From End of List

30 篇文章 0 订阅

做关于链表类的题目,经常提示 空指针 ,导致每次写程序,都会写if()去判断

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        if((head == NULL)) return NULL;
        int list_length = 0;
        ListNode *temp_head = head;
        while(temp_head != NULL){
            list_length++;
            temp_head = temp_head->next;
        }
        if(n == 0) return head;//没有删除元素
        if(n == list_length){// 删除头元素,如果元素只有一个,返回的是空链表。如果元素大于1,那么删除头节点
            temp_head = head;
            if(temp_head->next != NULL){
                temp_head = temp_head->next;
                return temp_head;
            }else{
                return NULL;
            }
        }
        
        temp_head = head;
        n = n%list_length;
        for(int i = 1;i<list_length-n;i++)
            temp_head = temp_head->next;
        if(n == 1){
            if(temp_head != NULL)
            {
                temp_head->next = NULL;   
            } 
        }else if(n>1){
            if((temp_head->next != NULL)&&(temp_head->next->next != NULL))
            temp_head->next = temp_head->next->next;
        }
        return head;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值