Merge pull request #12159 from deadmarshal/TWC325

TWC325

1 files changed, 21 insertions, 0 deletions

diff --git a/challenge-325/deadmarshal/java/Ch2.java b/challenge-325/deadmarshal/java/Ch2.java
new file mode 100644
index 0000000000..353b6f6506
--- /dev/null
+++ b/challenge-325/deadmarshal/java/Ch2.java
@@ -0,0 +1,21 @@
+import java.util.Arrays;
+import java.util.Stack;
+
+public class Ch2 {
+  public static void main(String[] args) {
+    System.out.println(Arrays.toString(final_price(new int[]{8, 4, 6, 2, 3})));
+    System.out.println(Arrays.toString(final_price(new int[]{1, 2, 3, 4, 5})));
+    System.out.println(Arrays.toString(final_price(new int[]{7, 1, 1, 5})));
+  }
+
+  private static int[] final_price(int[] arr) {
+    Stack<Integer> stack = new Stack<>();
+    for (int i = 0; i < arr.length; ++i) {
+      while (!stack.isEmpty() && arr[stack.peek()] >= arr[i])
+        arr[stack.pop()] -= arr[i];
+      stack.push(i);
+    }
+    return arr;
+  }
+}
+