题目:给定链表的头指针和一个结点指针,在O(1)时间删除该结点。链表结点的定义如下:
struct ListNode
{
int m_nKey;
ListNode* m_pNext;
};
函数的声明如下:
void DeleteNode(ListNode* pListHead, ListNode* pToBeDeleted);
方法来自于网络,完整代码自己抠的。
我们通常删除链表某个节点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; }