bzoj4236JOIOJI

bzoj4236JOIOJI

题意：

给一个只由JOI三个字母组成的串，求最长的一个子串使其中JOI三个字母出现次数相等。串长度≤200000

题解：

有点像bzoj4384，因此推算的过程是差不多的，但还是有不同因为本题要求的是出现次数相等，而那题要求的是不等：

cnt[1][i]-cnt[1][j]==cnt[2][i]-cnt[2][j]，cnt[2][i]-cnt[2][j]==cnt[3][i]-cnt[3][j]，cnt[1][i]-cnt[1][j]==cnt[3][i]-cnt[3][j]

化简得到cnt[1][i]-cnt[2][i]==cnt[1][j]-cnt[2][j]，cnt[2][i]-cnt[3][i]==cnt[2][j]-cnt[3][j]，cnt[1][i]-cnt[3][i]==cnt[1][j]-cnt[3][j]（本式实际上是冗余的因为可由前两式相加得到）

故可以用一个map维护二元组<cnt[1][i]-cnt[2][i],cnt[2][i]-cnt[3][i]>的最早出现次数i，每次在map中查找键值中有没有和当前元素的<cnt[1]-cnt[2],cnt[2]-cnt[3]>相等的元素，有的话和答案比较，否则插入map。

代码：

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <map>
 #define maxn 200100
 #define inc(i,j,k) for(int i=j;i<=k;i++)
 using namespace std;
 
 int a,b,c,d,e,n,ans; char s[maxn];
 map<pair<int,int>,int>m;
 int main(){
     scanf("%d%s",&n,s+); m[make_pair(,)]=;
     inc(i,,n){
         if(s[i]=='J')a++; if(s[i]=='O')b++; if(s[i]=='I')c++; d=a-b; e=b-c;
         if(m.find(make_pair(d,e))==m.end())m[make_pair(d,e)]=i;
         else ans=max(ans,i-m[make_pair(d,e)]);
     }
     printf("%d",ans); return ;
 }

20160814