剑指offer面试题13——在O(1)时间删除链表结点

题目:

给定单向链表的头指针和一个结点指针,定义一个函数在 O(1)时间删除该结点。链表结点与函数的定义如下:

剑指offer面试题13——在O(1)时间删除链表结点_第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;
}

思路:

如果删除的节点不是尾结点,那么将它下一个节点的值复制到它身上,然后删除它的下一个节点;

如果删除的节点是尾结点,那么从头结点开始遍历,遍历到它的前一个节点,然后删除尾结点;

如果删除的节点是尾结点且它又是头结点,那么直接将头结点置空。

你可能感兴趣的:(leetcode和机试题)