算法数组之种花问题

176 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第26天,点击查看活动详情

题目

假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。

给你一个整数数组  flowerbed 表示花坛,由若干 0 和 1 组成,其中 0 表示没种植花,1 表示种植了花。另有一个数 n ,能否在不打破种植规则的情况下种入 n 朵花?能则返回 true ,不能则返回 false。

Example

输入:flowerbed = [1,0,0,0,1], n = 1
输出:true

输入:flowerbed = [1,0,0,0,1], n = 2
输出:false

思路

采用跳格子的方式来解答

1.如果遍历的时候遇到1则当前索引已经种过花了,于是需要跳俩格才有可能种,i+2;

2.如果当前索引的后一位(i+1)遇到了0,由于之前我们跳格子都是跳俩格,则 i-1也必为0,当前索引就满足种花;如果当前索引的元素是最后的元素,由于之前跳格子是跳2格,i-1是0,而且当前最后一格,所以也满足种花,继续第一步的操作跳俩格,然后可种花的数量-1;

3.如果前索引的后一位(i+1) 遇到了1,则需要跳三格才能找到能种花的位置,也就是 i+3;

4.最后如果鲜花的数量小于等于0则表示可以种,否则就不能。

代码实现

class Solution {
    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        for (int i = 0; i < flowerbed.length;) {
            // 如果当前遇到1,则说明i+1一定是为0的,于是跳俩格
            if (flowerbed[i] == 1) {
                i += 2;
            // 由于遇到1都是跳俩格,并且根据规定i-1一定是为0,如果当到达最后一位时,由于后面已经没有值了,所以是可种的
            // 或者,如果当 i 下一位不为1时,则i处也可中
            } else if (i == flowerbed.length - 1 || flowerbed[i+1] == 0){
                n--;
                i += 2;
            // 否则当 i+1 处为1时,根据规则必须要跳三格才能看能不能种花
            } else {
                i += 3;
            }
        }

        return n <= 0;
    }
}

我是杰少,如果您觉的我写的不错,那请给我 点赞+评论+收藏 后再走哦!

往期文章:

请你喝杯 ☕️ 点赞 + 关注哦~

  1. 阅读完记得给我点个赞哦,有👍 有动力
  2. 关注公众号--- HelloWorld杰少,第一时间推送新姿势

最后,创作不易,如果对大家有所帮助,希望大家点赞支持,有什么问题也可以在评论区里讨论😄~**