Create Partition Array According to Given Pivot.java

Lokesh598 · May 24, 2024 · d27dd4a · d27dd4a
1 parent 2a1606b
commit d27dd4a
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/Array/Partition Array According to Given Pivot.java b/Array/Partition Array According to Given Pivot.java
@@ -0,0 +1,24 @@
+class Solution {
+    public int[] pivotArray(int[] nums, int pivot) {
+        int[] res = new int[nums.length];
+
+        int pointer = 0;
+        for (int i = 0; i < nums.length; i++) {
+            if (nums[i] < pivot) {
+                res[pointer++] = nums[i];
+            }
+        }
+        for (int i = 0; i < nums.length; i++) {
+            if (nums[i] == pivot) {
+                res[pointer++] = nums[i];
+            }
+        }
+        for (int i = 0; i < nums.length; i++) {
+            if (nums[i] > pivot) {
+                res[pointer++] = nums[i];
+            }
+        }
+
+        return res;
+    }
+}