HackerEarth Friendship and diseases problem solution - TheCScience