Java Specify Array Size Python Interview Coding Practice