给定链表的头指针和一个结点指针,在O(1)时间删除该结点

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

struct ListNode

{

      int        m_nKey;

      ListNode*  m_pNext;

};

函数的声明如下:

void DeleteNode(ListNodepListHeadListNodepToBeDeleted);

方法来自于网络,完整代码自己抠的。

我们通常删除链表某个节点node的时候,首先要找到这个节点。

常用的方法是使用2个指针q和pre,q在前面遍历链表查找node,而pre指向q刚遍历过的前一个节点。当q==node的时候,表示已经找到要删除的节点node了。

那么只要:pre->next = q->next; delete q;即可。

但是上面的方法复杂度显然是O(n),不符合题目的要求。


既然题目已经给我们指向待删除节点的指针,那么我们不需要遍历链表查询这个节点了。我们选择的方法是node节点的下一个节点p的值赋值到node节点的数据域中,而删除p节点,这样就从效果上说,就相当于删除了Node节点。

但是这样做有一个特例,就是要删除的是最后一个节点。此时node的下一个节点p是NULL。所以这时候我们还是要按常规办法,遍历链表,找到node前面的一个节点。

下面是完整可运行的代码

// HelloWorld.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
using namespace std;

typedef struct node{
	int data;
	struct node *next;
}Node,*pNode;

pNode createList(int n)
{
	pNode head = (pNode)malloc(sizeof(Node));
	head->data = 0;//head头指针的数据域存储的是链表的长度
	head->next = NULL;
	pNode p = head;
	int num;
	while(n--)
	{
		cout<<"input "<<n<<" node : ";
		cin>>num;
		pNode q = (pNode)malloc(sizeof(Node));
		q->data = num;
		q->next = NULL;
		p->next = q;
		p = q;
		(head->data)++;
	}
	return head;
}

void display(pNode head)
{
	if(head ==NULL)
	{
		cout<<"empty list"<<endl;
		return;
	}
	pNode p = head->next;
	cout<<p->data; 
	p = p->next;
	while(p)
	{
		cout<<"->"<<p->data;
		p = p->next;
	}
	cout<<endl;
}
pNode find(pNode head,int index)//找到链表中第index个节点,返回指向该节点的指针
{
	if(head == NULL) return NULL;
	if(index>head->data)
	{
		cout<<"index exceed the boundary!"<<endl;
		return NULL;
	}

	pNode p = head;
	while(p&&index--)
	{
		p = p->next;
	}
	return p;
}

void deleteNode(pNode head,pNode p)
{
	if(head==NULL || p==NULL)
		return;
	if(p->next!=NULL)//如果删除的不是最后一个节点
	{
		pNode q = p->next;
		p->data = q->data;
		p->next  = q->next;
	}else//如果删除的是最后一个节点
	{
		pNode r = head->next;
		while(r->next!=NULL && r->next!=p)
		{
			r = r->next;
		}
		r->next = p->next;
	}
}
int main()
{	int N = 5;
	pNode head = createList(N);//建立一个长度为N的链表
	display(head);//输出链表
	pNode p = find(head,1);//找到第n的节点
	deleteNode(head,p);//删除第n个节点
	display(head);//输出删除节点后的效果
	getchar();
	getchar();
	return 0;
}


你可能感兴趣的:(给定链表的头指针和一个结点指针,在O(1)时间删除该结点)