【链表】19. 删除链表的倒数第 N 个结点

2021-01-25  本文已影响0人  Spring_java

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

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


图片描述

这个题目的难度在于。测试样例
输入: [1]
输出:[]

使用快慢指针

public ListNode removeNthFromEnd(ListNode head, int n) {
       if(head==null){
            return null;
        }
        ListNode node1=head;
        ListNode node2=head;
        while(n>0){
            node1=node1.next;
            n--;
        }
        while( node1!=null && node1.next!=null){
            node1=node1.next;
            node2=node2.next;
        }
        node2.next=node2.next.next;
        return head;
    }

这里存在问题,当只有一个节点,并且要删除头节点的时候,会报错,不满足条件,
因为node2.next 是null。

解决方案:

增加虚拟的头节点,也使用快慢指针

ListNode pre = new ListNode(0);
        pre.next = head;
        ListNode start = pre, end = pre;
        while(n != 0) {
            start = start.next;
            n--;
        }
        while(start.next != null) {
            start = start.next;
            end = end.next;
        }
        end.next = end.next.next;
        return pre.next;
上一篇 下一篇

猜你喜欢

热点阅读