题目:
给你一个链表,删除链表的倒数第 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]
题解
方法一://倒数第n个,也就正数第length() - n + 1个,这我倒是想到了,就是忘了怎么找到第n个节点了。// 定义一个新节点用来指向结果链表的头节点,这个我也想到了。
我没有想到使用while来找到链表的长度,使用for循环找到第n个节点的长度。
package test;
class ListNode {
int val;
ListNode next;
ListNode(int x) {
val = x;
}
}
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
// 定义一个新节点用来指向结果链表的头节点。
ListNode node = new ListNode(0);//定义一个新节点不要忘了幅值0
// 将新节点指向当前链表的头节点。
node.next = head;
// 定义长度 len 来记录原链表的长度。
int len = 0;
// 定义 p 指针遍历链表,每有一个节点则长度 +1 。
ListNode p = head;//这里p就是在头结点,不要指向下一个极点了
while (p != null) {
len++;
p = p.next;
}
// 倒数第 n 个节点就是正数 len - n 的下一个节点。
len = len - n;
p = node;
// 遍历原链表找到第 len - n 个节点。
for (int i = 0; i < len; i++) {
p = p.next;
}
// 删除原链表第 len - n 的下一个节点。
p.next = p.next.next;
return node.next;
}
}
public class Main{
public static void main (String []args){
ListNode l1 = new ListNode(1);
ListNode l2 = new ListNode(2);
ListNode l3 = new ListNode(3);
ListNode l4 = new ListNode(4);
ListNode l5 = new ListNode(5);
l1.next = l2;
l2.next = l3;
l3.next = l4;
l4.next = l5;
Solution p = new Solution();
ListNode b = p.removeNthFromEnd(l1, 4);
System.out.println("结果:"+b.val);
}
}
方法二:双指针
整体思路是让前面的指针先移动n步,之后前后指针共同移动直到前面的指针到尾部为止
首先设立预先指针 pre,预先指针是一个小技巧
设预先指针 pre 的下一个节点指向 head,设前指针为 start,后指针为 end,二者都等于 pre
start 先向前移动n步
之后 start 和 end 共同向前移动,此时二者的距离为 n,当 start 到尾部时,end 的位置恰好为倒数第 n 个节点
因为要删除该节点,所以要移动到该节点的前一个才能删除,所以循环结束条件为 start.next != null
删除后返回 pre.next,为什么不直接返回 head 呢,因为 head 有可能是被删掉的点
时间复杂度O(n).
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
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;
}
}