Pergunta de entrevista da empresa ByteDance

Two coding questions, one easy and one medium