203、 移除链表元素
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
示例 1:
输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]
示例 2:
输入:head = [], val = 1
输出:[]
示例 3:
输入:head = [7,7,7,7], val = 7
输出:[]
提示:
列表中的节点数目在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= val <= 50
方法一:迭代
1.1 思路分析
首先要掌握的是链表的删除。用 temp \textit{temp} temp 表示当前节点。如果 temp \textit{temp} temp的下一个节点不为空且下一个节点的节点值等于给定的 val \textit{val} val,则需要删除下一个节点。删除下一个节点可以通过以下做法实现:
temp.next = temp.next.next
如果当前节点的下一节点等于val,使用上个等式对其进行删除;
如果当前节点的下一节点不等于val,则temp = temp.next;
head是头结点,可能会被删除,所以定义一个哑元节点指向head
1.2 代码实现
/**
* Definition for singly-linked list.
* 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; }
* }
*/
class Solution {
public ListNode removeElements(ListNode head, int val) {
ListNode dumpyHead = new ListNode(-1);
dumpyHead.next = head;
ListNode tmp = dumpyHead;
while (tmp.next != null){
if (tmp.next.val == val){
tmp.next = tmp.next.next;
}else {
tmp = tmp.next;
}
}
return dumpyHead.next;
}
}
1.3 测试结果
1.4 复杂度
- 时间复杂度:O(n)
- 空间复杂度:O(1)
方法二:递归
2.1 思路分析
链表的定义具有递归的性质,因此链表的题通常可以用递归来解决。
对于给定的链表,先对除头结点以外的节点进行操作,然后再判断头结点是否等于val;如果等于则删除head,返回head.next。否则返回head。
结束条件是当前节点head为空。
2.2 代码实现
/**
* Definition for singly-linked list.
* 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; }
* }
*/
class Solution {
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.3 测试结果
2.4 复杂度
- 时间复杂度:O(n)
- 空间复杂度:O(n)