[CodeForces - 873B] Balanced Substring【前缀和(01串) + map】

题意

• 给一个长度为n的01串，问'0'和'1'个数相同的子串中子串的最大长度

思路

• 将01串中的0看作'-1'，然后跑一遍for循环统计前缀和。那么如果两个位置ij的前缀和相同，则说明区间[i + 1, j]是一个'0'和'1'个数相同的子串.
• 那么我们只将未出现过的前缀和用map存起来，于是再次遇到存在于map中的前缀和则可以直接统计答案
• 注意，初始时需要将mp[0] = 0. 否则，如果整个串是'0'和'1'个数相同的，答案将偏小

---

这题本来正常统计的前缀和，用的二分，调了很久都不对。也想不明白为啥不对。但是又个样例是可以推翻二分的正确性的

12
110000110000
//ans = 8. 但是二分跑出来是4，
//因为第一次判mid = 6，但是没有长度为6的子串是合法的，于是区间缩小，否定大于等于6的子串。
//显然是不正确的。 

 

---

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <map>
#include <cstring>
#define INF 0x3f3f3f3f3f
using namespace std;

typedef long long ll;

inline int read()
{
    int x = 0, f = 1; char c = getchar();
    while(c < '0' || c > '9') { if(c == '-') f = -f; c = getchar(); }
    while(c >= '0' && c <= '9') { x = x * 10 + c - '0'; c = getchar(); }
    return x * f;
}

const int maxN = 100005;

int n, prefix[maxN];
char s[maxN];
map<int, int>mp;
map<int, int>:: iterator it;

int main()
{
    n = read();
    scanf("%s", s);
    for(int i = 0; i < n; ++ i)
    {
        if(s[i] == '0') prefix[i + 1] = prefix[i] - 1;
        else prefix[i + 1] = prefix[i] + 1;
    }
    int ans = 0;
    mp[0] = 0;
    for(int i = 1; i <= n; ++ i )
    {
        int now = prefix[i];
        it = mp.find(now);
        if(it != mp.end())
            ans = max(ans, i - it->second);
        else
            mp[now] = i;
    }
    printf("%d\n", ans);
    return 0;
}
/*
6
110001
*/