题目:给你一个整数数组 nums 和一个链表的头节点 head。从链表中移除所有存在于 nums 中的节点后,返回修改后的链表的头节点。
思路:把数组元素放到set里,用set.contains判断,其他部分 按照模板写即可
代码:
- /**
- * 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 modifiedList(int[] nums, ListNode head) {
- Set
set = new HashSet<>(); - for (int num : nums)
- set.add(num);
- ListNode dummy = new ListNode(0, head);
- ListNode cur = dummy;
- while (cur.next != null) {
- int value = cur.next.val;
- if (set.contains(value)) {
- cur.next = cur.next.next;
- } else {
- cur = cur.next;
- }
- }
- return dummy.next;
- }
- }
性能:
时间复杂度o(m+n)m链表长度 n是数组长度
空间复杂度o(n)