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

Posted 2021-10-03 潜行前行

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

进阶：你能尝试使用一趟扫描实现吗？

示例 1：

输入：head = [1,2,3,4,5], n = 2
输出：[1,2,3,5]
示例 2：

输入：head = [1], n = 1
输出：[]
示例 3：

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

题解

/**
* 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) {
       ListNode backNode = head;
       ListNode tmp = head , pre = head;
       for(int i=1;i<n;i++){
           backNode = backNode.next;
       }
       boolean first = true;
       while(backNode.next !=null){
           backNode = backNode.next;
           tmp = tmp.next;
           if(first){
               first = false;
           }else{
               pre = pre.next;
           }
       }
       if(head == tmp){
           head = head.next;
       }else{
           pre.next = tmp.next;
       }
       return head;
   }
}