LeetCode热题100道-Day01

11,792 阅读1分钟

LeetCode热题100道-Day01

605. 种花问题

  • 假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。给你一个整数数组  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;

}