Domanda di colloquio di ByteDance

Is there a faster way to implement this solution?