来自PoPoQQQ大爷的题目以及题解。。。传送门
事实证明虽然常数吓人,但是善于利用各种STL还是能带来很大的方便,比如这道题。
我们记表示前i个字符中J、O、I分别出现的次数,那么一个从j+1到i的字串满足条件的充要条件是,
化简得,所以我们开一个map<pair <int ,int>,int>,记录这两个差值第一次出现的位置,一边扫一边求以该点作为结尾的答案即可。
/**************************************************************
Problem: 4236
User: RicardoWang
Language: C++
Result: Accepted
Time:748 ms
Memory:4828 kb
****************************************************************/
#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
using namespace std;
#define MAXN 200005
map<pair<int,int>,int> m;
int N,A[MAXN];
void Init()
{
scanf("%d",&N); char op;
for(int i=1;i<=N;i++)
{
op=getchar(); while(op!='J' && op!='O' && op!='I')op=getchar();
A[i]= op=='J' ? 0 : ( op=='O' ? 1 : 2 );
}
return ;
}
void work()
{
int c[3],ans=0;
pair<int,int>t; t=make_pair(0,0);
m[t]=0;
c[0]=c[1]=c[2]=0;
for(int i=1;i<=N;i++)
{
c[A[i]]++;
t=make_pair(c[0]-c[1],c[1]-c[2]);
if(m.find(t)==m.end())
{
m[t]=i;
}
else
{
ans=max(ans,i-m[t]);
}
}
printf("%d\n",ans);
return ;
}
int main()
{
// freopen("in.txt","r",stdin);
Init();
work();
return 0;
}