移除链表元素(无傀儡节点,递归+迭代)
Posted 小写丶H
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了移除链表元素(无傀儡节点,递归+迭代)相关的知识,希望对你有一定的参考价值。
问题
一个单链表(无傀儡节点/头结点),删除指定的所有元素。
如 :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;
//写进循环里,挑出条件为cur为null时。
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);
// if(head.val==val){
// return head.next;
// }else{
// return head;
// }
//简便写法
return head.val == val ? head.next : head;
}
递归代码很简洁,但是(较抽象)需要理解,但是如果链表很长,那么不建议使用,防止StackOverflowError(栈溢出问题)。
以上是关于移除链表元素(无傀儡节点,递归+迭代)的主要内容,如果未能解决你的问题,请参考以下文章