Python Correlation Between Two Arrays Codeforces