删除链表中指定的元素

cdtft · May 12, 2020 · 2f4420f · 2f4420f
1 parent 0abf78f
commit 2f4420f
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/main/java/com/cdtft/leetcode/Solution.java b/src/main/java/com/cdtft/leetcode/Solution.java
@@ -8,6 +8,7 @@
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Stack;
 import java.util.TreeMap;
 
@@ -450,4 +451,25 @@ public int rob(int[] nums) {
         return dp1;
     }
 
+    public ListNode removeElements(ListNode head, int val) {
+        while (head != null && head.val == val) {
+            head = head.next;
+        }
+        if (head == null) {
+            return head;
+        }
+        ListNode index = head;
+        while (!Objects.isNull(index.next)) {
+            if (index.next.val == val) {
+                index.next = index.next.next;
+                continue;
+            }
+            index = index.next;
+            if (index == null) {
+                return head;
+            }
+        }
+        return head;
+    }
+
 }
diff --git a/src/test/java/com/cdtft/leetcode/SolutionTest.java b/src/test/java/com/cdtft/leetcode/SolutionTest.java
@@ -115,4 +115,10 @@ public void rob() {
         int[] nums = new int[]{1, 2, 3, 1};
         System.out.println(solution.rob(nums));
     }
+
+    @Test
+    public void removeElements() {
+        ListNode node = new ListNode(1);
+        solution.removeElements(null, 1);
+    }
 }