sherlock and squares hackerrank solution in python