203.给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
- 拿到题目以为是头结点不存元素的链表,一顿操作发现过不了,审题很重要!题目有给图示要注意看。然后一顿修改发现又错了,在第二个循环之前就提前将第一个指针p设置成head(这没毛病),然后将指针q设置成head->next,这里忽略了第一个循环后head就可能是NULL了,NULL值的next是什么呢?这我也不知道,所以就报错了。按照题解又修改了一遍,最终得到如下一种在原链表上修改的解法。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* removeElements(struct ListNode* head, int val){
struct ListNode * p, * q;
//删除符合条件的头结点
while (head != NULL && head->val == val) {
p = head;
head = head->next;
free(p);
}
//删除符合条件的非头结点
p = head;
while (p != NULL && p->next != NULL) {
if (p->next->val == val) {
q = p->next;
p->next = p->next->next;
free(q);
} else {
p = p->next;
}
}
return head;
}
- 上面这种解法需要分别对头结点和非头结点写一段逻辑来处理,而如果给这个链表增加一个头结点,那么原链表所有的结点都可以用一种逻辑来处理了,其实这也是我第一次做错时的写法,但是没想到还能增加一个虚拟头结点这种操作。注意在结尾时要重新给head指针赋值,因为原先head指针指向的结点可能已经被释放掉了。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* removeElements(struct ListNode* head, int val){
struct ListNode* dummyHead = (struct ListNode*)malloc(sizeof(struct ListNode));
dummyHead->next = head;
struct ListNode * p, * q;
//删除符合条件的非头结点
p = dummyHead;
while (p != NULL && p->next != NULL) {
if (p->next->val == val) {
q = p->next;
p->next = p->next->next;
free(q);
} else {
p = p->next;
}
}
head = dummyHead->next;
free(dummyHead);
return head;
}