https://alandroidnet.com/related/chocolate stack hackerearth solution python