相等长度01串与01序列

前缀和

#include <cstdio>
#include <algorithm>
#include <iostream>

using namespace std;
#define MAXNUM 111111
#define rep(i,s,t) for(int i=s;i<t;i++)
#define pii pair<int,int>
typedef long long ll;
int sum[MAXNUM*2];char s[MAXNUM];
int main()
{
    int num0=0,num1=0,res1=0,res2=0;
    int n;
    scanf("%d%s",&n,s);int snow=0;
    rep(i,0,n)
    {
        if(s[i]=='0')num0++,snow+=-1;
        else num1++,snow++;
        if(!sum[snow+n])sum[snow+n]=i;
        if(!snow)
            res1=max(res1,i+1);
        else 
            res1=max(res1,i-sum[snow+n]);
    }
    res2=min(num0,num1)*2;
    printf("%d %d\n",res1,res2);
}