首页 > 代码库 > [leetcode]_Remove Nth Node From End of List

[leetcode]_Remove Nth Node From End of List

2024-07-04 17:28:30   229人阅读

题目：移除linked-list从尾到头的第N个元素

自我思路：因为题目给出的N是从链表尾开始计算的，单链表不存在从子指向父亲的反向指针，因此我先计算链表的整个长度len，然后用len - N来表示正向被删除元素所在的位置。

代码：

public ListNode removeNthFromEnd(ListNode head, int n) {
        if(head == null) return null;
        
        ListNode help = head;
        int length = 1;
        while(help.next != null){
            length++;
            help = help.next;
        }
        
        if(n == length){
            head = head.next;
        }else{
            ListNode temp = head;
            for(int i = 1 ; i < length - n ; i++) temp = temp.next;
            
            if(n == 1) temp.next = null;
            else temp.next = temp.next.next;
        }
        return head;
    }

这样的算法有个很低效的地方，如果我要删除一个very very 长的链表的倒数第二个元素，我得先遍历一遍，这一步可能半小时过去了。然后我又得查找一遍，才能删除。

 

网络上的思路非常精辟，双指针（快指针 & 慢指针），虽然原理还没有很理解，但是举例这样的确能得到正确答案。快指针比慢指针先走N步，N步以后快慢指针同时继续走，当快指针走到链表尾部时，慢指针指向的就是被删除元素的前一个元素。精妙的思路。

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

 


联系
我们