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