大体思路
题目链接:https://www.luogu.com.cn/problem/P9952
这题的二分很一眼。感觉有些题解的代码过于复杂了。
考虑二分答案,如果可以插入两头牛则 l = m i d + 1 l=mid+1 l=mid+1,并更新 a n s ans ans,否则 r = m i d − 1 r=mid-1 r=mid−1。
如何判定我们二分的 m i d mid mid 是否符合要求呢?
我们设 a i a_i ai 表示第 i i i 个已经有牛的牛栏的位置。容易发现,只要 a i + x + x a_i + x+ x ai+x+x 还小于等于 a i + 1 a_{i+1} ai+1,那么在中间显然可以加一头牛。注意,两个已经有牛的牛栏之间可能可以把 2 2 2 头牛都加进去,所以,安排完一头牛就直接更新 a a a 数组。
然后注意一下把牛安排在边界的情况就可以了。
代码如下:
#include <bits/stdc++.h>
using namespace std;
#define int long long
typedef long long ll;
typedef pair<int, int> pr;
#define up(i, l, r) for(int i = (l); i <= (r); i++)
#define down(i, r, l) for(int i = (r); i >= (l); i--)
const int mod = 1000000007;
const int base = 2333;
const double eps = 1e-6;
inline int read() {
int x = 0, f = 1; char ch = getchar();
while(ch < '0' || ch > '9') { if(ch == '-') f = -1; ch = getchar(); }
while(ch >= '0' && ch <= '9') { x = (x << 1) + (x << 3) + (ch ^ 48); ch = getchar(); }
return x * f;
}
int n, m, k, Q, T, _, ans = 0;
bool c[100007];
int a[100007], cnt = 0;
int b[100007], bcnt;
int l = 1, r = 2e9, mid;
inline bool check(int x) {
int num = 0; bcnt = cnt;
for(int i = 1; i <= cnt; i++) b[i] = a[i];
b[++bcnt] = 1 - x;
b[++bcnt] = n + x;
sort(b + 1, b + bcnt + 1);
for(int i = 1; i <= bcnt; i++) {
if(b[i] + x + x <= b[i + 1]) {
num++;
if(num == 1) {
b[++bcnt] = b[i] + x;
sort(b + 1, b + bcnt + 1);
}
if(num == 2) {
return 1;
}
}
}
return 0;
}
signed main() {
n = read();
for(int i = 1; i <= n; i++) {
char ch; cin >> ch;
c[i] = ch - '0';
if(c[i] == 1) {
a[++cnt] = i;
if(cnt >= 2) r = min(r, a[cnt] - a[cnt - 1]);
}
}
while(l <= r) {
mid = (l + r) / 2;
if(check(mid)) l = mid + 1, ans = mid;
else r = mid - 1;
}
cout << ans;
return 0;
}