C. Left and Right Houses

Problem - C - Codeforces

题目分析

<1>0：想被分割至左边； 1：想被分割至右边

<2>使得左右两侧均有一半及其以上的人满意（我*******）

<3>答案若有多个，取最接近中间位置的答案

<4>使用前缀和确定到每个位置时有多少个 “1”； 再遍历前缀和数组；此时可以求出满意人数

代码：

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 3e5 + 9;

char a[N];

int main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int t; cin >> t;
    while(t--)
    {
        int n; cin >> n; 
        string s; cin >> s;
        s = ' ' + s + '0'; //前缀和预处理
        int f[n + 5] = {0};
        for(int i = 1; i <= n + 1; i++) //房屋两边也能铺路
        {
            f[i] = f[i - 1] + (s[i] - '0');
        }
        
        
        int p = -1;
        for(int i = 0; i <= n + 1; i++)
        {
            int l = i - f[i]; //左边满意人数
            int r = f[n + 1] - f[i]; //右边满意人数
            
            if(2 * l >= i && r * 2 >= n - i)
            {
                if(abs(n - 2 * i) < abs(n - 2 * p)) p = i;
            }
        }
        cout << p << '\n';
    }
}