http://coolshell.cn/articles/8990.html
Linus举了一个单向链表的例子,但给出的代码太短了,一般的人很难搞明白这两个代码后面的含义。正好,有个编程爱好者阅读了这段话,并给出了一个比较完整的代码。他的话我就不翻译了,下面给出代码说明。
如果我们需要写一个remove_if(link*, rm_cond_func*)的函数,也就是传入一个单向链表,和一个自定义的是否删除的函数,然后返回处理后的链接。
这个代码不难,基本上所有的教科书都会提供下面的代码示例,而这种写法也是大公司的面试题标准模板:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
typedef
struct
node
{
struct
node * next;
....
} node;
typedef
bool
(* remove_fn)(node
const
* v);
// Remove all nodes from the supplied list for which the
// supplied remove function returns true.
// Returns the new head of the list.
node * remove_if(node * head, remove_fn rm)
{
for
(node * prev = NULL, * curr = head; curr != NULL; )
{
node *
const
next = curr->next;
if
(rm(curr))
{
if
(prev)
prev->next = next;
else
head = next;
free
(curr);
}
else
prev = curr;
curr = next;
}
return
head;
}
|
这里remove_fn由调用查提供的一个是否删除当前实体结点的函数指针,其会判断删除条件是否成立。这段代码维护了两个节点指针prev和curr,标准的教科书写法——删除当前结点时,需要一个previous的指针,并且还要这里还需要做一个边界条件的判断——curr是否为链表头。于是,要删除一个节点(不是表头),只要将前一个节点的next指向当前节点的next指向的对象,即下一个节点(即:prev->next = curr->next),然后释放当前节点。
但在Linus看来,这是不懂指针的人的做法。那么,什么是core low-level coding呢?那就是有效地利用二级指针,将其作为管理和操作链表的首要选项。代码如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
void
remove_if(node ** head, remove_fn rm)
{
for
(node** curr = head; *curr; )
{
node * entry = *curr;
if
(rm(entry))
{
*curr = entry->next;
free
(entry);
}
else
curr = &entry->next;
}
}
|
同上一段代码有何改进呢?我们看到:不需要prev指针了,也不需要再去判断是否为链表头了,但是,curr变成了一个指向指针的指针。这正是这段程序的精妙之处。(注意,我所highlight的那三行代码)
让我们来人肉跑一下这个代码,对于——
1)(第12行)如果不删除当前结点 —— curr保存的是当前结点next指针的地址。
2)(第5行) entry 保存了 *curr —— 这意味着在下一次循环:entry就是prev->next指针所指向的内存。
3)(第8行)删除结点:*curr = entry->next; —— 于是:prev->next 指向了 entry -> next;
是不是很巧妙?我们可以只用一个二级指针来操作链表,对所有节点都一样。
如果你对上面的代码和描述理解上有困难的话,你可以看看下图的示意:
#include
#include
using namespace std;
struct node {
int a;
node *next;
};
void print(node* head) {
for(;head!=NULL;head = head->next) {
cout << head->a << " ";
}
cout << endl;
}
//this method is brief
node* rm(node* head,int val) {
node *pre = NULL;
node *curr = head;
for (;curr != NULL;) {
node *next = curr->next;
if(curr->a%2 == val) {
if (pre == NULL) {
head = next;
}
else {
pre->next = next;
}
free(curr);
}
else {
pre = curr;
}
curr = next;
}
return head;
}
void remove(node **head, int val) {
//curr指向的是当前节点的指针的指针
//对于中间节点而言,当前节点的指针和其前一个节点的next是一样的,所以可以看做是前一个节点的next的指针
for (node** curr = head; *curr != NULL;) {
node *entry = *curr;//entry是当前节点的指针//1. curr是头节点,那么*curr = entry->next,就是指head指向了下一个节点//2.如果curr是中间节点,则curr是前一个节点next的指针
if (entry->a % 2 == val) {
*curr = entry->next;
free(entry);
}
else {
curr = &(entry->next);
}
}
}
int main() {
node *head = (node*)malloc(sizeof(node));
head->a = 1;
node *a = (node*)malloc(sizeof(node));
a->a = 2;
node *b = (node*)malloc(sizeof(node));
b->a = 3;
node *c = (node*)malloc(sizeof(node));
c->a = 4;
node *d = (node*)malloc(sizeof(node));
d->a = 5;
head->next = a;
a->next = b;
b->next = c;
c->next = d;
d->next = NULL;
print(head);
remove(&head,0);
print(head);
return 1;
}
给还不明白的同学论证一下,见如下代码
typedef struct list_s{
struct list_s * next;
int a;
int b;
} list_t;
main()
{
list_t lA, lB, lC, lD;
list_t ** phead = &lA.next;
lA.next = &lB;
lA.a = 11;
lA.b = 12;
lB.next = &lC;
lB.a = 21;
lB.b = 22;
lC.next = &lD;
lC.a = 31;
lC.b = 32;
lD.next = NULL;
lD.a = 41;
lD.b = 42;
printf(“%d\n”, ((list_t *)phead)->a);
printf(“%d\n”, (**(list_t**)phead).a);
printf(“%d\n”, (***(list_t***)phead).a);
printf(“%d\n”, (****(list_t****)phead).a);
printf(“%d\n”, ((list_t *)phead)->b);
printf(“%d\n”, (**(list_t**)phead).b);
printf(“%d\n”, (***(list_t***)phead).b);
printf(“%d\n”, (****(list_t****)phead).b);
}
gcc的结果是
11
21
31
41
12
22
32
42