删除链表中倒数第n个结点
问题描述
给你一个链表,删除链表的倒数第 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;
}
}