CF873B Balanced Substring(前缀和)

看到01串计算长度，显然是前缀和，进行贡献转化，把0看成-1，再用map

但是我犯了个错误，我虽然存了0这个位置，但是因为0的位置是0，我在后面使用m1[sum]来查找，其实是查不到的，所以要用count函数来找

#include<iostream>
#include<cstring>
#include<cstdio>
#include<map>
#include<algorithm>
#define ull unsigned long long
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int N=1e5+10;
int sum;
map<int,int> m1;
int main(){
    int n;
    cin>>n;
    string s;
    cin>>s;
    int i;
    s=" "+s;
    m1[0]=0;
    int ans=0;
    for(i=1;i<=n;i++){
        if(s[i]=='0')
            sum--;
        else
            sum++;
        if(m1.count(sum))
            ans=max(ans,i-m1[sum]);
        else
            m1[sum]=i;
    }
    cout<<ans<<endl;


}

View Code