如何在 VSCode 中通过 CodeGeeX 优化现有的算法逻辑

CodeGeeX 这次更新的上下文感知重构，可以说把算法优化的门槛又拉低了一截。以前你要手动改一段二分查找、快排分区或者 DP 状态转移，总得在简洁性、鲁棒性和语言特性之间反复权衡，现在它直接在代码上下文中给出语义等价的重构方案，省去了大量手动试错的时间。 具体怎么用？四种方式，挑顺手的来。

## 右键触发上下文感知重构 选中你要优化的函数体、循环块或者条件分支区域——确保光标在那片选区里——然后右键，选择「Refactor with CodeGeeX」（IntelliJ 里可能叫「Optimize with CodeGeeX」）。等个 1～3 秒，面板弹出 2～4 个候选方案。 这时候你可以勾选“提取边界检查为 guard clause”或者“用三元运算替代单行 if-else”，然后点 Apply。如果原始代码存在空指针风险，插件会自动插入 `Objects.nonNull()` 或 `Optional.ofNullable()` 封装。注意，这步不会改你原有逻辑，只是补全 null 安全校验。 ## 用注释指令驱动定向优化 方法一：在算法代码上方写一条自然语言注释，比如： `// 将该递归斐波那契改为带 memo 的迭代实现，避免重复计算，空间复杂度控制在 O(1)` 然后光标放在注释末尾，按 Alt+（Windows/Linux）或 Option+（Mac），从建议列表里选一个插入。 方法二：如果你想保留递归结构但提升可读性，注释可以写成： `// 用 @lru_cache 修饰该函数，显式声明缓存大小为 128，并为参数添加类型提示` 触发后检查生成的代码是否真的包含了 `@lru_cache(maxsize=128)` 和 `def fib(n: int) -> int`。 这里有个关键点：注释里必须出现具体约束词，比如“O(1)”“maxsize=128”“类型提示”。否则 CodeGeeX 可能只会给你一个泛泛的简化版本，达不到预期。 ## 分步验证并注入测试用例 第一步，先让 CodeGeeX 解释当前算法的最坏时间复杂度和关键瓶颈点，确认它理解正确； 第二步，粘贴三个典型测试输入——比如空数组、全相同元素、逆序数组——然后要求：“生成的优化版本必须通过以下全部用例，否则重写”； 第三步，对双指针类算法，额外追加一句：“当 left == right 时，确保不越界访问，且返回值语义与原版一致”。 这一步千万别省略。CodeGeeX 默认生成的优化代码很可能忽略极端索引场景，手动注入测试用例相当于强制模型反向推导边界防护逻辑。 ## 侧边栏发起多维度优化请求 点击 VS Code 右侧的 CodeGeeX 图标打开面板，输入指令：“请从可读性、时间复杂度、Python 3.9+ 特性适配三个维度评估以下代码，并分别给出优化建议”。然后粘贴原始算法，发送。 它会返回结构化的反馈：比如“可读性：变量名 idx 可改为 insertion_point；时间复杂度：当前为 O(n²)，建议改用 bisect 模块；特性适配：可用 match-case 替代长 if-elif 链”。每条建议都附带可直接运行的代码片段，复制粘贴就能生效。