题目描述:
给定单向列表的头指针和一个节点指针,定义一个函数在O(1)时间删除该节点。
解题思路:
一般单向链表删除节点,就是从头遍历,找到这个节点,指向这个节点之后的节点,就算是删除了这个节点。
这种算法的复杂度为O(n)。
题目要求在O(1)时间内完成,所以应该避免遍历。
我们遍历的目的是为了找到这个节点前面的那个节点,所以我们可以采用下面的方法。
将待删除节点的下一个节点的值赋给待删除的节点,然后删除一个节点,就达到了删除节点的目的。
找到合适的算法后,还有几种情况需要考虑:
1)正常,多个节点,删除的不是尾节点。
2)只有一个节点,删除尾节点(也是头节点)。
3)有多个节点,删除尾节点。
代码如下:
class ListNode {
int data;
ListNode next;
}
public class DeleteNode {
public ListNode deleteNode(ListNode head, ListNode deListNode) {
if(deListNode == null || head == null)
return null;
if(deListNode.next != null) {//不是尾节点
ListNode node = deListNode.next;
deListNode.data = node.data;
deListNode.next = node.next;
node = null;
} else if(head == deListNode) {//只有一个节点,删除头节点
deListNode = null;
head = null;
return null
} else {//多个节点,删除尾节点
ListNode pointListNode = head;//从头结点开始
while(pointListNode.next != deListNode)
pointListNode = pointListNode.next;
pointListNode.next = null;
deListNode = null;
}
return head;
}
}