Solve array and tree algorithm challenges | TikTok Interview Question