刷题的日常-种花问题

58 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第18天,点击查看活动详情

刷题的日常-2022年8月13日

一天一题,保持脑子清爽

种花问题

来自leetcode的 605 题,题意如下:

假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。
给你一个整数数组flowerbed 表示花坛,由若干 0 和 1 组成,其中 0 表示没种植花,1 表示种植了花。另有一个数n ,能否在不打破种植规则的情况下种入n朵花?能则返回 true ,不能则返回 false。

示例如下:

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

理解题意

我们可以从题意中提取的条件如下:

  • 题目给出一个数组
  • 数组的元素只有0 和 1
    • 0代表没种花
    • 1代表已经种了花
  • 要求我们往0的位置种花
  • 相邻的位置不能种花
  • 返回最大的种植数量

做题思路

说白了就是填空题,要我们往没有种植的地方放入花,放进去的时候检查下是否两边都没有花就可以了,其实就是一种贪心的思想:

  • 遍历数组,检查每个元素
  • 如果已经种植了,就跳过
  • 如果没有种植,就需要检查下当前位置是否左右两边都没有被种植
  • 如果两边都是空的,意味着当前位置可以放
  • 统计已经种下的数量
  • 返回最终结果即可

代码实现

代码实现如下:

public class Solution {

    public boolean canPlaceFlowers(int[] flowerbed, int n) {
        boolean left, right;
        for (int i = 0; i < flowerbed.length; i++) {
            if (flowerbed[i] == 1) {
                continue;
            }
            left = checkLeft(flowerbed, i);
            right = checkRight(flowerbed, i);
            if (left && right) {
                flowerbed[i] = 1;
                n--;
                if (n == 0) {
                    return true;
                }
            }
        }

        return n <= 0;
    }

    private boolean checkRight(int[] flowerbed, int i) {
        return i == flowerbed.length - 1 || flowerbed[i + 1] == 0;
    }

    private boolean checkLeft(int[] flowerbed, int i) {
        return i == 0 || flowerbed[i - 1] == 0;
    }

}