题目描述
思路
由于题目要求: Could you do this in one pass?
所以这题的难点在于:
- 我们不通过获取链表长度的方式来拿需要删除的节点。
要定位到需要删除的点,我们可以这样操作,假设我们要删除的点是X,会有如下关系
其中:
-
X'点是X关于中点的对称点
-
Y是X前一个点
-
Y'是X’后一个点
所以,可以很容易的得到一个结论:
如果我设置两个指针,假设一个叫pre,一个叫cur,
-
cur从头开始走,当cur走到Y'位置的时候(即走了n+1步),pre开始从头走;
-
接下来cur和pre都继续向下一个位置移动,当cur移动到结尾(cur == null)的时候,pre应该正好在Y位置;
-
pre.next = pre.next.next;
这样就把X位置给删除掉了。
这道题另外一个需要注意的点是:
- 需要被删除的节点如果不是头节点,那么返回的链表节点应该是头节点。
- 如果需要删除的节点正好是头节点,那么返回的节点应该是头节点的下一个节点。
所以,如果上述两个指针移动的第一步:
cur从头开始走,当cur走到Y'位置的时候(即走了n+1步),正好是cur == null的位置。那pre不用走了,头节点即为需要删除的节点。
直接返回 head.next即可。
完整代码:
public static ListNode removeNthFromEnd(ListNode head, int n) {
ListNode cur = head;
ListNode pre = null;
while (cur != null) {
n--;
if (n == -1) {
pre = head;
}
if (n < -1) {
pre = pre.next;
}
cur = cur.next;
}
if (pre == null) {
return head.next;
}
pre.next = pre.next.next;
return head;
}