[LEET-3354] add 3354

fishercoder1534 · fishercoder1534 · commit fa75bc2ffa56 · 2024-11-22T07:56:16.000-08:00
diff --git a/paginated_contents/algorithms/4th_thousand/README.md b/paginated_contents/algorithms/4th_thousand/README.md
@@ -1,5 +1,6 @@
 | #    | Title                                                                                                                                                              | Solutions                                                                                                                         | Video                                                                         | Difficulty | Tag                                                                  
 |------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------|-------------------------------------------------------------------------------|------------|----------------------------------------------------------------------
+| 3354 | [Make Array Elements Equal to Zero](https://leetcode.com/problems/make-array-elements-equal-to-zero/)                                                        | [Java](https://github.com/fishercoder1534/Leetcode/blob/master/src/main/java/com/fishercoder/solutions/fourththousand/_3354.java) |                                                                               | Easy       |
 | 3349 | [Adjacent Increasing Subarrays Detection I](https://leetcode.com/problems/adjacent-increasing-subarrays-detection-i/)                                                        | [Java](https://github.com/fishercoder1534/Leetcode/blob/master/src/main/java/com/fishercoder/solutions/fourththousand/_3349.java) |                                                                               | Easy       |
 | 3340 | [Check Balanced String](https://leetcode.com/problems/check-balanced-string/)                                                        | [Java](https://github.com/fishercoder1534/Leetcode/blob/master/src/main/java/com/fishercoder/solutions/fourththousand/_3340.java) |                                                                               | Easy       |
 | 3330 | [Find the Original Typed String I](https://leetcode.com/problems/find-the-original-typed-string-i/)                                                        | [Java](https://github.com/fishercoder1534/Leetcode/blob/master/src/main/java/com/fishercoder/solutions/fourththousand/_3330.java) |                                                                               | Easy       |
diff --git a/src/main/java/com/fishercoder/solutions/fourththousand/_3354.java b/src/main/java/com/fishercoder/solutions/fourththousand/_3354.java
@@ -0,0 +1,51 @@
+package com.fishercoder.solutions.fourththousand;
+
+import java.util.Arrays;
+
+public class _3354 {
+    public static class Solution1 {
+        public int countValidSelections(int[] nums) {
+            int count = 0;
+            for (int i = 0; i < nums.length; i++) {
+                if (nums[i] == 0) {
+                    if (isValidWithMoveDirection(nums, i, true)) {
+                        count++;
+                    }
+                    if (isValidWithMoveDirection(nums, i, false)) {
+                        count++;
+                    }
+                }
+            }
+            return count;
+        }
+
+        private boolean isValidWithMoveDirection(int[] nums, int index, boolean moveLeft) {
+            int[] copy = Arrays.copyOf(nums, nums.length);
+            while (index >= 0 && index < nums.length) {
+                if (moveLeft) {
+                    if (copy[index] > 0) {
+                        copy[index]--;
+                        moveLeft = !moveLeft;
+                        index++;
+                    } else {
+                        index--;
+                    }
+                } else {
+                    if (copy[index] > 0) {
+                        copy[index]--;
+                        moveLeft = !moveLeft;
+                        index--;
+                    } else {
+                        index++;
+                    }
+                }
+            }
+            for (int num : copy) {
+                if (num != 0) {
+                    return false;
+                }
+            }
+            return true;
+        }
+    }
+}
diff --git a/src/test/java/com/fishercoder/fourththousand/_3354Test.java b/src/test/java/com/fishercoder/fourththousand/_3354Test.java
@@ -0,0 +1,26 @@
+package com.fishercoder.fourththousand;
+
+import com.fishercoder.solutions.fourththousand._3354;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+public class _3354Test {
+    private _3354.Solution1 solution1;
+
+    @BeforeEach
+    public void setup() {
+        solution1 = new _3354.Solution1();
+    }
+
+    @Test
+    public void test1() {
+        assertEquals(2, solution1.countValidSelections(new int[]{1, 0, 2, 0, 3}));
+    }
+
+    @Test
+    public void test2() {
+        assertEquals(3, solution1.countValidSelections(new int[]{16, 13, 10, 0, 0, 0, 10, 6, 7, 8, 7}));
+    }
+}
