给定单向链表的头结点以及一个待删除结点,定义一个函数在
O(1)
时间内删除该结点。
思路
这一天nowcoder上面没有,就没有OJ测试了,主要是学习一下删除结点 的另外一种思路。
(1)一般删除链表中的某个结点的做法都是遍历到该结点的前面,此时时间复杂度就是O(n)
了,然后就是常规的curNode.next=delNode.next;delNode.next=null;size--
完事。
(2)由于本题的要求是O(1)
时间复杂度,故肯定不能从head开始遍历的这种方法。采用的是复制结点法delNode.val=delNode.next.val;delNode.next=delNode.next.next;delNode.next=null
。
(3)注意特殊情况,只有一个结点的时候;以及delNode是尾结点时需要遍历O(n)
。整体时间复杂度[O(1)*(n-1)+O(n)]/n=O(1)
。
(4)关于复制结点本质就是复制值和引用,因此other
写法也是OK;另外,关于head = head.next
最好不要直接使用,建议写成ListNode temp=head;temp = temp.next
,最初的头结点不要动,复制一份使用,方便后面再次使用头结点。
解法
public ListNode deleteOneNode(ListNode head,ListNode delNode){
//error input
if (head==null || delNode==null){
return null;
}
//delNode is the first one
if (head.next==null){
head=null;
return head;
}
ListNode curNode=head;
//delNode is the last one
if (delNode.next==null){
while (curNode.next!=delNode){
curNode=curNode.next;
}
curNode.next=null;
delNode=null;
return head;
}
//generally
curNode=delNode.next;
delNode.val=curNode.val;
delNode.next=curNode.next;
curNode.next=null;
//other
// curNode=delNode.next;
// delNode=curNode;
// curNode.next=null;
return head;
}