offer2-21


给定一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。

示例 1:

img

输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]

示例 2:

输入:head = [1], n = 1
输出:[]

示例 3:

输入:head = [1,2], n = 1
输出:[1]

提示:

  • 链表中结点的数目为 sz
  • 1 <= sz <= 30
  • 0 <= Node.val <= 100
  • 1 <= n <= sz

进阶:能尝试使用一趟扫描实现吗?


//leetcode submit region begin(Prohibit modification and deletion)
/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode removeNthFromEnd(ListNode head, int n) {
        if (head == null || n <= 0) {
            return head;
        }

        ListNode fast = head;
        ListNode slow = null;
        int cnt = 0;

        while (fast != null) {
            fast = fast.next;
            if (cnt >= n) {
                if (slow == null) {
                    slow = head;
                } else {
                    slow = slow.next;
                }
            }
            cnt++;
        }

        if (slow == null) {
            if (cnt == n) {
                ListNode newHead = head.next;
                head.next = null;
                return newHead;
            } else {
                return head;
            }
        } else {
            ListNode remove = slow.next;
            slow.next = remove.next;
            remove.next = null;
            return head;
        }
    }
}
//leetcode submit region end(Prohibit modification and deletion)

文章作者: 倪春恩
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 倪春恩 !