劍指offer面試題13——在O（1）時間刪除鏈表結點

題目：

給定單向鏈表的頭指針和一個結點指針，定義一個函數在 O（1）時間刪除該結點。鏈表結點與函數的定義如下：

代碼：

void DeleteNode(ListNode** pListHead,ListNode* pToBeDeleted) {
	if (!pListHead || !pToBeDeleted)return;
	ListNode* p = pToBeDeleted;
	if (p->next) {
		p->val = p->next->val;
		p->next = p->next->next;
	}
	else {
		ListNode* head = *pListHead;
		if (!head->next) {
			*pListHead = NULL;
		}
		else {
			while (head->next && head->next->next) {
				head = head->next;
			}
			head->next = NULL;
		}
	}
}

int main() {
	ListNode* l1 = new ListNode(1);
	ListNode* l2 = new ListNode(2);
	ListNode* l3 = new ListNode(3);
	ListNode* l4 = new ListNode(4);
	ListNode* l5 = new ListNode(5);
	
	l2->next = l3;
	l3->next = l4;
	l4->next = l5;

	ListNode** pListHead = &l1;
	ListNode* pToBeDeleted = l1;
	DeleteNode(pListHead, pToBeDeleted);

	cout << pListHead;
	return 0;
}

思路：

如果刪除的節點不是尾結點，那麼將它下一個節點的值複製到它身上，然後刪除它的下一個節點；

如果刪除的節點是尾結點，那麼從頭結點開始遍歷，遍歷到它的前一個節點，然後刪除尾結點；

如果刪除的節點是尾結點且它又是頭結點，那麼直接將頭結點置空。