diff --git a/java/separate-black-and-white-balls.java b/java/separate-black-and-white-balls.java new file mode 100644 index 0000000..1e2bcbc --- /dev/null +++ b/java/separate-black-and-white-balls.java @@ -0,0 +1,19 @@ +class Solution { + public long minimumSteps(String s) { + long swaps = 0; + + int blackBalls = 0; + for (int i = 0; i < s.length(); ++i) { + switch (s.charAt(i)) { + case '0': + swaps += blackBalls; + break; + case '1': + ++blackBalls; + break; + } + } + + return swaps; + } +}