AcWing 1659. 社交距离 I

AcWing 1659. 社交距离 I

#include <iostream>

using namespace std;

int n,m,ans;
string str;
int location[100010];//记录牛的位置

bool check(int num) {
    int add=0;
    add+=location[0]/num+(n-location[m-1]-1)/num;//两边的牛的情况
    if(add>=2) return true;//可能不止两种情况，满足则返回
    //中间的牛的情况
    for(int i=0;i<m-1;i++){
        add+=(location[i+1]-location[i])/num-1;
    }
    return add>=2;
}

int main() {
    cin >> n;
    int r=n,l=1;
    cin >> str;
    for(int i=0; i<n; i++) {
        if(str[i]=='1')
            location[m++]=i;//记录第m+1个牛的位置
    }
    if(m==0) {
        cout << n-1;
        return 0;
    }
    for(int i=1;i<m;i++)
        r=min(r,location[i]-location[i-1]);//最小距离，结果一定在[1,r]之间
    //二分法求答案
    while(l<=r)
    {
        int mid=(r+l)/2;
        if(check(mid)) {
            ans=mid;
            l=mid+1;
        }
        else
            r=mid-1;
    }
    cout << ans;
    return 0;
}