LeetCode热题100道-Day01
- 假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。给你一个整数数组 flowerbed 表示花坛,由若干 0 和 1 组成,其中 0 表示没种植花,1 表示种植了花。另有一个数 n ,能否在不打破种植规则的情况下种入 n 朵花?能则返回 true ,不能则返回 false。
bool canPlaceFlowers(vector<int>& flowerbed, int n) {
unordered_map<int, int> res;// 0/1,count
res[0] = 0;
res[1] = 0;
size_t i = 0;
while (i < flowerbed.size() && flowerbed[i] == 0 ) {
res[0] ++;
i++;
}
if (i == flowerbed.size()) {//全是 0
return (res[0] + 1) / 2>=n;
}
int plantFlower = res[0] / 2;//遇到1了,此时flowerbed[i] = 1
res[1] = 1;
res[0] = 0;
i++;
for (; i < flowerbed.size();++i ) {
res[flowerbed[i]] += 1;
if (res[1] == 2) {
//countZero = res[0];
plantFlower += (res[0] - 1) / 2;
res[1] = 1;
res[0] = 0;
}
}
if (res[1] == 2) {
//countZero = res[0];
plantFlower += (res[0] - 1) / 2;
}
else if (res[1] == 1) {
//countZero = res[0];
plantFlower += res[0] / 2;
}
return plantFlower >= n;
}