I am trying to convert this Python Solution in Java. For some reason, my Java Solution is not working. How can this be done correctly?