NC53 删除链表的倒数第n个节点

  算法   3分钟   697浏览   0评论

题目链接:https://www.nowcoder.com/practice/f95dcdafbde44b22a6d741baf71653f6

题目描述

给定一个链表,删除链表的倒数第 n 个节点并返回链表的头指针
例如,给出的链表为: 1→2→3→4→5, n= 2.
删除了链表的倒数第 n 个节点之后,链表变为1→2→3→5.

数据范围: 链表长度 0≤n≤1000,链表中任意节点的值满足 0≤val ≤100

要求:空间复杂度 O(1),时间复杂度 O(n)
备注:题目保证 n 一定是有效的

示例 1:

输入:{1,2},2
返回值:{2}

解题代码

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param head ListNode类 
     * @param n int整型 
     * @return ListNode类
     */
    public ListNode removeNthFromEnd (ListNode head, int n) {
        // write code here
        if (head == null || n == 0) return head;
        ListNode cur = head, pre = head;
        for (int i = 0; i < n; i++) {
            if (cur == null) return head;
            cur = cur.next;
        }
        if (cur == null) {
            return head.next;
        }
        while (cur.next != null) {
            cur = cur.next;
            pre = pre.next;
        }
        pre.next = pre.next.next;
        return head;
    }
}

如果你觉得文章对你有帮助,那就请作者喝杯咖啡吧☕
微信
支付宝
  0 条评论