Nearby Squares | Hackerearth solution