新兵队列训练算法 Python 实现 - 优化后的代码和解释

def queue_training(n, soldiers):\n result = []\n for num in soldiers:\n queue = [i+1 for i in range(num)]\n while len(queue) > 3:\n queue = queue[2:] + queue[:2]\n queue = [i for i in queue if (i+1) % 3 != 0]\n result.append(queue)\n return result\n\nn = int(input())\n soldiers = []\n for _ in range(n):\n soldiers.append(int(input()))\n\n result = queue_training(n, soldiers)\n for queue in result:\n print(*queue)