问题
一个单链表(无傀儡节点/头结点),删除指定的所有元素。 如 :1->3->5->7->5->9 删除5之后;变为:1->3->7->9
思路
可以用递归,但是空间复杂度高。 所以使用迭代。遍历一次链表便删除所有指定节点。 使用两个两个指针 cur 和 prve 。
基本删除思路如图片所示。
还要考虑一些细节,没有判断头结点是否删除。所有写到代码注释里面了。
代码
如果代码没懂,多画图理解一下就会明白。
迭代
public ListNode removeElements(ListNode head, int val) {
if(head==null){
return null;
}
ListNode cur=head.next;
ListNode prve=head;
while(cur!=null){
if(cur.val==val){
prve.next=cur.next;
cur=cur.next;
}else{
prve=prve.next;
cur=cur.next;
}
}
if(head.val==val){
head=head.next;
}
return head;
}
如果只是删除第一次出现的元素,只需要把循环去掉微改一下就可以。
递归
public ListNode removeElements(ListNode head, int val) {
if (head == null) {
return head;
}
head.next = removeElements(head.next, val);
return head.val == val ? head.next : head;
}
递归代码很简洁,但是(较抽象)需要理解,但是如果链表很长,那么不建议使用,防止StackOverflowError(栈溢出问题)。
|