题意: 给定一个长度为 n n n的 01 01 01序列,问在给定的 01 01 01序列中可以使得多少个 0 0 0变成 1 1 1,其中任意两个 1 1 1不相邻。
数据范围:可以出成 O ( n ) O(n) O(n)
题解:
先考虑一个全 0 0 0的序列:
对于偶数:放置 1 , 3 , 5 , 7 , . . . , 2 k + 1 1,3,5,7,...,2k+1 1,3,5,7,...,2k+1或者 2 , 4 , 6 , 8 , . . . , n 2,4,6,8,...,n 2,4,6,8,...,n
对于奇数:放置 1 , 3 , 5 , . . . , n 1,3,5,...,n 1,3,5,...,n
贡献即 ⌈ n + 1 2 ⌉ \lceil \frac{n+1}{2}\rceil ⌈2n+1⌉。
再考虑对于任意一个 01 01 01序列:
问题转换为了带有首字符和尾字符放置限制的 01 01 01序列。
我们处理的必然是其中每一个全 0 0 0子串:长度为 l e n len len
- 如果是一个首尾都被限制的子串,贡献为: ⌈ l e n − 2 + 1 2 ⌉ \lceil \frac{len-2+1}{2}\rceil ⌈2len−2+1⌉
- 如果是一个首字符被限制的,即该子串的尾字符为原序列的最后一个字符,贡献为: ⌈ l e n − 1 + 1 2 ⌉ \lceil \frac{len-1+1}{2}\rceil ⌈2len−1+1⌉
- 如果是一个尾字符被限制的,即该子串的首字符为原序列的第一个字符,贡献为: ⌈ l e n − 1 + 1 2 ⌉ \lceil \frac{len-1+1}{2}\rceil ⌈2len−1+1⌉
- 如果是一个首尾字符都不被限制的,即为全 0 0 0序列,贡献为 ⌈ l e n + 1 2 ⌉ \lceil \frac{len+1}{2}\rceil ⌈2len+1⌉
代码:
class Solution {
public:
bool canPlaceFlowers(vector<int>& f, int n) {
int cnt = 0, m = f.size();
for(int i = 0; i < m; ++i) {
if(f[i]) continue;
int fir = i, en = i;
while(en < m && f[en] == 0) ++en;
--en;
int len = en - fir + 1;
if(fir > 0 && en < m - 1) cnt += len - 1 >> 1;
else if((fir > 0 && en == m - 1) || (fir == 0 && en < m - 1)) cnt += len >> 1;
else cnt += len + 1 >> 1;
i = en;
}
return n <= cnt;
}
};