嵌套循环与transform实现的两段相似C++代码输出差异原因排查

你的原代码内层循环完全达不到预期效果，核心问题是循环过程中修改了依赖的基准值，具体拆解如下：

原循环的执行逻辑错误 #

看这段内层循环：

for(int j = i; j < n; j++){
    dup[j] -= dup[i];
}

当j=i时，第一次执行dup[i] -= dup[i]，直接把dup[i]改成了0。后续j=i+1、i+2...的循环中，dup[j] -= dup[i]实际上是减0，等于完全没修改后面的元素。最终只有dup[i]被置为0，其余元素毫无变化——这就和你移除内层循环的效果几乎一致（毕竟dup[i]之后也不会再被处理）。

transform版本为什么正确 #

你用的transform代码：

transform(dup.begin() + i, dup.end(), dup.begin() + i,
                         [value = dup[i]](int x) { return x - value; });

这里的关键是lambda表达式的value = dup[i]是值捕获：在lambda创建的瞬间，就把当前dup[i]的原始值保存到了value变量里，后续所有元素的减法操作都是基于这个固定的原始值，不会因为dup[i]被修改而变化。

修正原循环的方法 #

只需要提前把dup[i]的原始值保存下来，再执行循环即可：

if (dup[i] != 0){
    ans += (n - i + 1) * dup[i];
    int val = dup[i]; // 先保存基准值
    for(int j = i; j < n; j++){
        dup[j] -= val;
    }
}

示例验证 #

假设dup数组为[2,2,3]，当i=0时：

• 原循环执行后：dup变成[0,2,3]，仅第一个元素被修改
• 修正后的循环/transform执行后：dup变成[0,0,1]，所有后续元素都减去了原始的2，符合逻辑预期

内容的提问来源于stack exchange，提问作者AyushRaj