Pregunta de entrevista de TikTok

Write code to implenment the Sort the flags problem/