Bash Script Array Size Python Interview Coding Question