删除链表中倒数第n个结点_双指针法_java

删除链表中倒数第n个结点

leetcode链接

问题描述

给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。你能尝试使用一趟扫描实现吗？

提示：

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

测试用例

示例 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]

解题思路

• 为使每次交换元素时的操作一致，创建一个虚拟头结点
• 使用双指针法。使快指针比慢指针先行n步，然后再让快慢指针循环指向下一个元素，直到快指针指向null。此时，慢指针就指向要删除结点的前一个结点。最后进行节点删除。

代码实现

/**
* 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 {
   /*
   时间复杂度：O(n)
   空间复杂度：O(1)
   */
   public ListNode removeNthFromEnd(ListNode head, int n) {
       //创建虚拟头结点，便于对链表各个结点的统一操作
       ListNode dumyHead = new ListNode (-1, head);
       ListNode slow = dumyHead;
       ListNode fast = head;
       //令快指针比慢指针先行n步
       while (n != 0) {
           fast = fast.next;
           n--;
       }
       //当快指针指向null时，慢指针就指向要删除结点的前一个结点
       while (fast != null) {
           slow = slow.next;
           fast = fast.next;
       }
       //删除倒数第n个结点
       slow.next = slow.next.next;
       //注意：若删除的是头结点则head需要指向新的头结点！
       head = dumyHead.next;

       return head;
   }
}