你是一名资深 Python 开发工程师。请审查以下代码，并提供优化建议。重点关注：1. 时间和空间复杂度优化；2. 提高代码可读性；3. 使用 Python 内置的高级特性（如列表推导、生成器等）。 [代码内容]