思路
这个题以前刷宝典的时候做过
忘了思路了。看了眼提示。才想起来。
手写一遍AC
需要注意两点:
- 这里考虑到表头可能被删除,我就加了个统计表长的变量。
- 考虑到摘除节点,需要维护所删除节点的前一个节点
- 删除的是表头节点,则直接返回表头节点;删除的是其他节点,摘除节点后返回head即可
代码
/**
* Definition for a singly-linked list.
* class ListNode {
* public $val = 0;
* public $next = null;
* function __construct($val) { $this->val = $val; }
* }
*/
class Solution {
/**
* @param ListNode $head
* @param Integer $n
* @return ListNode
*/
function removeNthFromEnd($head, $n) {
$p1 = $head;
$p2 = $head;
$step = $n;
$count = 0;
while ($step > 0) {
$p1 = $p1->next;
$count++; //统计链表长度,特殊处理链表头结点被删除的情况
$step--;
}
$pre = $head;
while ($p1 != null) {
$p1 = $p1->next;
$pre = $p2;
$p2 = $p2->next;
$count++;
}
if ($count == $n) {
return $head->next;
} else {
$pre->next = $p2->next;
$p2->next = null;
return $head;
}
}
}