Create Partition Array According to Given Pivot.java

Author: Lokesh598

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;
+    }
+}