参考:https://blog.csdn.net/biezhihua/article/details/79889987 ,写的特别好。以下是我的实现:
题目
给定一个链表,删除链表的倒数第 n 个节点并返回头结点。
例如,
给定一个链表: 1->2->3->4->5, 并且 n = 2.
当删除了倒数第二个节点后链表变成了 1->2->3->5.
1
2
3
说明:
给的 n 始终是有效的。
尝试一次遍历实现。
解法
很简单的一道题,既然只允许遍历一次,且N一直有效,那么可以利用双指针解法(前指针、后指针),让前指针先走N步,再让两个在指针同时后移,直到前指针到达尾部,此时,后指针的下一个节点就是要被删除的节点了。
@Test
public void test() {
ListNode head = new ListNode(1);
ListNode listNode = removeNthFromEnd(head, 1);
System.out.println(listNode);
}
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode preNode = head;
ListNode curNode = head;
for (int i = 0; i < n; i++) {
curNode = curNode.next;
}
if (curNode == null) {
return preNode.next;
}
while (curNode.next != null) {
preNode = preNode.next;
curNode = curNode.next;
}
preNode.next = preNode.next.next;
return head;
}
class ListNode {
int val;
ListNode next;
ListNode(int x) {
val = x;
}
}