Array Halves Codechef Python