我一开始的做法是想着暴力解题,但是看到了题目的进阶要求说能不能就遍历一次链表,我就放弃了这种做法;其次的想法是放入一个数组,然后再根据数组的序号进行删除操作,但是这样感觉把问题复杂化了
接着想到看了最近左神的算法课讲到的链表部分,做法一般分为两种:快慢指针和额外的辅助数据结构。所以我用了第二种方法:有序表。
思路就是如果有序表的key是基本类型,那么能够按照Key大小的顺序值进行排列,并能获取最大的key值对应value的功能。将链表的节点的序号作为key,节点本身作为value加入有序表中,然后再根据n的值取出对应的结点进行操作。
实现代码如下:
class Solution{
public ListNode removeNthFromEnd(ListNode head, int n) {
TreeMap<Integer,ListNode> tm = new TreeMap<>();
if (head.next == null){
return null;
}
ListNode p = head;
int v;
for (int i = 1;p.next!=null;i++){
tm.put(i,p);
v = p.val;
p = p.next;
}
int max = tm.lastKey();
if (max == n){
head = head.next;
return head;
}
ListNode q = tm.get(max-n);
q.next = tm.get(max-n+1).next;
return head;
}
}
最后虽然通过了,但是时间和空间复杂度并不理想:
于是去学习官方题解和其他人的优秀解法:
首先
在对链表进行操作时,一种常用的技巧是添加一个哑节点(dummy node),它的 next 指针指向链表的头节点。这样一来,我们就不需要对头节点进行特殊的判断了。
例如,在本题中,如果我们要删除节点 y,我们需要知道节点 y 的前驱节点 x,并将 x 的指针指向 y 的后继节点。但由于头节点不存在前驱节点,因此我们需要在删除头节点时进行特殊判断。但如果我们添加了哑节点,那么头节点的前驱节点就是哑节点本身,此时我们就只需要考虑通用的情况即可。
特别地,在某些语言中,由于需要自行对内存进行管理。因此在实际的面试中,对于「是否需要释放被删除节点对应的空间」这一问题,我们需要和面试官进行积极的沟通以达成一致。下面的代码中默认不释放空间。
然后使用的办法就是左神课程中提到的第二种方法:快慢指针
我们也可以在不预处理出链表的长度,以及使用常数空间的前提下解决本题。
由于我们需要找到倒数第 n 个节点,因此我们可以使用两个指针 first 和 second 同时对链表进行遍历,并且 first 比second 超前 nn 个节点。当 \textit{first}first 遍历到链表的末尾时,second 就恰好处于倒数第 nn 个节点。
具体地,初始时 first 和 second 均指向头节点。我们首先使用first 对链表进行遍历,遍历的次数为 n。此时,first 和second 之间间隔了 n−1 个节点,即irst 比 second 超前了 nn 个节点。
在这之后,我们同时使用 first 和second 对链表进行遍历。当 first 遍历到链表的末尾(即 first 为空指针)时,second 恰好指向倒数第 nn 个节点。
根据方法一和方法二,如果我们能够得到的是倒数第 nn 个节点的前驱节点而不是倒数第 n 个节点的话,删除操作会更加方便。因此我们可以考虑在初始时将second 指向哑节点,其余的操作步骤不变。这样一来,当 first 遍历到链表的末尾时,second 的下一个节点就是我们需要删除的节点。
实现代码如下:
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode dummy = new ListNode(0, head);
ListNode first = head;
ListNode second = dummy;
for (int i = 0; i < n; ++i) {
first = first.next;
}
while (first != null) {
first = first.next;
second = second.next;
}
second.next = second.next.next;
ListNode ans = dummy.next;
return ans;
}
}