在算法题中,处理数组的边界条件是一个常见的挑战。特别是在涉及多条件判断时,如何高效且清晰地处理边界问题,可以显著提升代码的简洁性和可读性。本文将以一道经典的算法题——花坛种花问题,来探讨边界条件的巧妙处理方法。
605. 种花问题 - 力扣(LeetCode)
给定一个由 0 和 1 组成的整数数组 flowerbed,0 表示该位置没有种花,1 表示该位置已经种花。花不能种在相邻的地块上,即不能有两个相邻的1。再给定一个整数 n,问是否能在不打破规则的情况下种入 n 朵花。
flowerbed = [1, 0, 0, 0, 1], n = 1trueflowerbed = [1, 0, 0, 0, 1], n = 2falseflowerbed,找出每一个可以种花的位置。i,我们检查它的前后位置(i-1和i+1)是否为空(即是否为0)。这里需要特别注意数组的边界情况: i是第一个位置,我们只需检查i+1。i是最后一个位置,我们只需检查i-1。n减1。n变为0,说明可以成功种植n朵花,返回true。如果遍历结束后n仍然大于0,返回false。在实现过程中,有一个非常精妙的边界条件处理方法,即通过如下判断来确保前后位置的检查不会越界:
bool emptyLeft = (i == 0) || (flowerbed[i - 1] == 0); bool emptyRight = (i == length - 1) || (flowerbed[i + 1] == 0); 这段代码通过简单的逻辑运算符,巧妙地处理了边界条件,避免了常见的越界错误。以下是详细的解释:
前位置检查:
i == 0:如果i是第一个位置,那么它没有前一个位置,所以直接视为前位置为空。flowerbed[i - 1] == 0:如果i不是第一个位置,检查i-1位置是否为空。后位置检查:
i == length - 1:如果i是最后一个位置,那么它没有后一个位置,所以直接视为后位置为空。flowerbed[i + 1] == 0:如果i不是最后一个位置,检查i+1位置是否为空。通常情况下,当我们处理数组的边界条件时,会采用如下方式:
if (i > 0 && flowerbed[i - 1] == 0) { // 处理前一个位置 } if (i < length - 1 && flowerbed[i + 1] == 0) { // 处理后一个位置 } 这种方式需要多次检查i是否越界,虽然能够保证程序的正确性,但显得繁琐且冗长。相比之下,上述巧妙的边界条件处理方式则显得更为简洁:
下面是完整的代码实现:
class Solution { public: bool canPlaceFlowers(vector& flowerbed, int n) { int cnt = 0; for (int i = 0; i < flowerbed.size(); i++) { if ((flowerbed[i] == 0) && (i == 0 || flowerbed[i - 1] == 0) && (i == flowerbed.size() - 1 || flowerbed[i + 1] == 0)) { flowerbed[i] = 1; cnt++; } if(cnt >= n){ return true; } } return cnt >= n; } }; 通过上述代码和分析,我们可以看到,巧妙的边界条件处理不仅让代码更简洁明了,而且减少了潜在的错误。希望这篇文章能帮助我们更好地理解边界条件的处理方法,并将这种技巧应用到其他算法问题中。