Unit Array Codeforces Solutions In Python