优化下面代码： n = int(input()) heights = list(map(int, input().split())) i = 0 while len(heights) > 1: if i + 1 >= len(heights): a = 0 else: a = i + 1 if heights[i] >= heights[a]: print(heights[a]) del heights[a] if a == 0: i = 0 i = i + 1 if i >= len(heights): i = 0

可以对代码进行如下优化：

n = int(input())  
heights = list(map(int, input().split()))  
i = 0

while len(heights) > 1:
    a = (i + 1) % len(heights)
    if heights[i] >= heights[a]:
        print(heights[a])
        del heights[a]
    else:
        i += 1

print(*heights, sep='\n')

优化说明：
- 使用取模操作 `(i + 1) % len(heights)` 可以简化判断下一个石头的索引，避免使用条件语句。
- 简化了 `if a == 0` 的判断，因为当 `i + 1 >= len(heights)` 时，取模操作 `(i + 1) % len(heights)` 已经保证了 `a` 的值为 0。
- 最后使用 `print(*heights, sep='\n')` 打印剩余的石头高度，避免在循环中多次打印。

请注意，优化代码可能会更加高效，但仍然保持了与原始代码相同的逻辑。