Pregunta de entrevista de Gokul Infocare

basic coding question and modifying code