问题描述:删除链表中等于给定值val的所有节点。
样例:给出链表 1->2->3->3->4->5->3,和val = 3,你需要返回删除3之后的链表:1->2->4->5。
// Definition for singly-linked list.
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
分析:首先要判断链表是否为空,是的话直接返回,然后考虑到单链表的删除需要两个指针变量的配合作用,那么就要来定义两个变量,一个
cur
保存当前节点,另一个
nex
保存当前节点的下一个节点,从链表的第二个节点开始遍历,若
nex
的值与
val
的值相等,则将
cur->next
指向
cur->next->next(
即
nex->next)
,删除节点。继续移动将
cur
和
nex
下移一位,往后遍历,直到链尾
(nex
为空
)
。再判断头结点
head
的
value
是否与
val
相等,如果相等,将头结点下移,删除原头结点,否则不做任何操作。遍历结束后返回
head
。
class Solution {
public:
/*
* @param head: a ListNode
* @param val: An integer
* @return: a ListNode
*/
ListNode * removeElements(ListNode * head, int val) {
// write your code here
if(head == NULL){
return head;
}
ListNode *cur = head;
ListNode *nex = head -> next;
while (nex != NULL){
if(nex -> val == val){
cur -> next = nex -> next;
nex = nex -> next;
}else{
cur = cur -> next;
nex = nex -> next;
}
}
if(head -> val == val) head = head -> next;
return head;
}
}; 注意:要解决该问题,首先要熟悉单链表节点的删除,我的第一反应时保存当前节点和上一节点,单链表中对上一节点的定位有些麻烦,本例中定义当前节点和下一节点的方法效率更高一些。
其次,对于头结点的判断,我刚开始觉得放在判断头结点是否为空之后也可以,试了一下发现报错了,例如原链表1->1->null和val为1,期望值null,结果1->null。出现的问题是,本例从第二个节点开始遍历,除了第一个节点外已经没有值与val相等的节点了,而提前判断之后再遍历,则会少判断第二个节点的值。
个人看法,如有错误,请批评指出。
|