题目描述:
LeetCode 605. Can Place Flowers
Suppose you have a long flowerbed in which some of the plots are planted and some are not. However, flowers cannot be planted in adjacent plots - they would compete for water and both would die.
Given a flowerbed (represented as an array containing 0 and 1, where 0 means empty and 1 means not empty), and a number n, return if n new flowers can be planted in it without violating the no-adjacent-flowers rule.
Example 1:
Input: flowerbed = [1,0,0,0,1], n = 1 Output: True
Example 2:
Input: flowerbed = [1,0,0,0,1], n = 2 Output: False
Note:
- The input array won't violate no-adjacent-flowers rule.
- The input array size is in the range of [1, 20000].
- n is a non-negative integer which won't exceed the input array size.
题目大意:
给定数组flowerbed表示一个花床,0表示位置为空,1表示位置非空。花不能相邻种植,即不能出现相邻的1。
给定想要种植的花朵数目n,判断是否可以满足要求。
解题思路:
贪心算法(Greedy Algorithm)
从左向右遍历flowerbed,将满足要求的0设为1。计数与n比较即可。
Python代码:
class Solution(object):
def canPlaceFlowers(self, flowerbed, n):
"""
:type flowerbed: List[int]
:type n: int
:rtype: bool
"""
ans = 0
for i, v in enumerate(flowerbed):
if v: continue
if i > 0 and flowerbed[i - 1]: continue
if i < len(flowerbed) - 1 and flowerbed[i + 1]: continue
ans += 1
flowerbed[i] = 1
return ans >= n
本文链接:http://bookshadow.com/weblog/2017/06/04/leetcode-can-place-flowers/
请尊重作者的劳动成果,转载请注明出处!书影博客保留对文章的所有权利。