Java 中 if 与 while 在滑动窗口收缩逻辑中的关键区别

本文详解为何在“最多翻转 k 个 0 得到最长连续 1”问题中,用 if 替代 while 会偶然通过测试但本质错误,而正确解法必须使用 while + math.max() —— 核心在于窗口收缩的完整性与答案更新的时机。

在滑动窗口算法中,if 与 while 的语义差异并非“执行次数不同”那么简单,而是直接决定了窗口是否真正满足约束条件(即 current ≤ k),进而影响 ans 的正确性。

先看原始 while 版本的问题所在——它逻辑正确但答案更新有缺陷

while (current > k) {
    if (nums[left] == 0) current--;
    left++;
}
ans = right - left + 1; // ❌ 错误:未取最大值!

该 while 循环本身完全正确:它持续收缩左边界,直到窗口内 0 的个数 current ≤ k,确保 [left, right] 始终是合法窗口。问题出在后续这行:

ans = right - left + 1;

它用当前合法窗口长度直接覆盖 ans,而非保留历史最大值。例如当窗口从长度 5 收缩为 4 后,ans 被错误地设为 4,丢失了之前更大的 5。正确做法是:

ans = Math.max(ans, right - left + 1); // ✅ 始终维护最大值

而 if 版本看似“修复”了问题,实则引入更隐蔽的错误:

if (current > k) {
    if (nums[left] == 0) current--;
    left++;
}
ans = right - left + 1; // ❌ 仍无 Math.max,且收缩不彻底!

此处 if 仅收缩一次左边界。但 current > k 时,可能需移动 left 多次才能使 current ≤ k(例如 k=1,当前窗口含 3 个 0)。此时 if 导致窗口非法(current 仍 > k),后续 right - left + 1 计算的是一个不满足题意的、过度膨胀的窗口长度——它依赖于“恰好一次收缩后碰巧合法”的巧合,因此在部分测试用例中结果偏大或偏小,属于未定义行为。

✅ 正确解法必须同时满足两点:

  • 使用 while (current > k) 确保窗口始终合法(收缩至 current ≤ k);
  • 使用 Math.max(ans, right - left + 1) 确保 ans 是所有合法窗口长度的最大值

完整正确代码如下:

class Solution {
    public int longestOnes(int[] nums, int k) {
        int current = 0;   // 当前窗口中 0 的个数
        int left = 0;
        int ans = 0;

        for (int right = 0; right < nums.length; right++) {
            if (nums[right] == 0) {
                current++;
            }

            // 关键:while → 彻底收缩,保证 [left, right] 合法
            while (current > k) {
                if (nums[left] == 0) {
                    current--;
                }
                left++;
            }

            // 关键:Math.max → 在所有合法窗口中取最长
            ans = Math.max(ans, right - left + 1);
        }
        return ans;
    }
}

⚠️ 注意事项:

  • 滑动窗口的“合法性”由 while 循环保障,不可简化为 if;
  • ans 必须在每次窗口合法后立即更新最大值,不能假设窗口长度单调递增;
  • 时间复杂度仍为 O(n):每个元素最多被 left 和 right 各访问一次。

总结:if 与 while 在此场景的本质区别,不是“执行一次还是多次”,而是“是否严格维持窗口约束”。算法正确性永远优先于表面简洁性。