给你一个链表的头节点 head
和一个整数 val
,请你删除链表中所有满足 Node.val == val
的节点,并返回 新的头节点 。
输入:head = [1,2,6,3,4,5,6], val = 6 输出:[1,2,3,4,5]
public class ListNode {
int val;
ListNode next;
ListNode() {}
ListNode(int val) { this.val = val; }
ListNode(int val, ListNode next) { this.val = val; this.next = next; }
自己写的实现:
public ListNode removeElements(ListNode head, int val) {
if( head == null){
return head;
}
if( head.next == null){
if(head.val == val){
return null;
}
return head;
}
if(head.val == val){
return removeElements(head.next,val);
}
return new ListNode(head.val,removeElements(head.next,val));
}
推荐解法1:
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;
}
推荐解法2:
public ListNode removeElements(ListNode head, int val) {
ListNode dummyHead = new ListNode(0);
dummyHead.next = head;
ListNode temp = dummyHead;
while (temp.next != null) {
if (temp.next.val == val) {
temp.next = temp.next.next;
} else {
temp = temp.next;
}
}
return dummyHead.next;
}
反思:
1.代码不够优雅,相比于推荐解法1,代码冗余。且使用了new,空间复杂度更高。
2.递归解法的时间复杂度为O(n),空间复杂度为O(n),所以推荐使用迭代的方式来实现。
场景:树结构如部门等信息的过滤可以考虑使用。