面试题13：在O(1)时间删除链表结点

题目：给定单向链表的头指针和一个结点指针，定义一个函数在O(1)时间删除该结点。

链表结点与函数的定义如下：

 struct ListNode

 {

 　　int val;

 　　ListNode* next;

 };

分析：如下图所示，假设要删除的结点为p结点，若直接删除p结点，则必须知道p的上一个结点，而查找p的上一个结点的时间复杂度为O(N)。可以交换p和p的下一个结点q的数据，然后直接删除q，这样便保证了时间复杂度为O(1)。需要注意的是p为尾结点的情况。这时只能顺序遍历链表并完成删除操作。

 ListNode * deleteNode( ListNode* head, ListNode* toBeDeleted)

 {

     if (head==NULL || toBeDeleted == NULL)

         return head;

     if (toBeDeleted->next!=NULL)

     {//删除的结点不是尾结点

         ListNode* q = toBeDeleted->next;

         toBeDeleted->val = q->val;

         toBeDeleted->next = q->next;

         delete q;

         q = NULL;

     }

     else

     {

         if (toBeDeleted == head)

         {//删除的结点既是头结点也是尾结点

             delete toBeDeleted;

             head = NULL;

             toBeDeleted = NULL;

         }

         else

         {//删除的结点是尾结点不是头结点

             ListNode* p = head;

             while (p->next!=toBeDeleted)

                 p = p->next;

             p->next = NULL;

             delete toBeDeleted;

             toBeDeleted = NULL;

         }

     }

     return head;

 }

PS：上述代码存在一个问题，因为它基于一个假设：要删除的结点的确在链表中。我们需要O(N)的时间才能判断链表中是否包含某一结点。

受到O(1)时间的限制，我们不得不把确保结点在链表中的责任推给了函数的调用者。在面试的时候，可以和面试官讨论这个假设，以便让面试官觉得我们考虑问题非常全面。

巴特西

面试题13：在O(1)时间删除链表结点

最新文章

热门文章