【BZOJ4236】JOIOJI [DP]

JOIOJI

Time Limit: 10 Sec  Memory Limit: 256 MB
[Submit][Status][Discuss]

Description

　　JOIOJI桑是JOI君的叔叔。“JOIOJI”这个名字是由“J、O、I”三个字母各两个构成的。

　　最近，JOIOJI桑有了一个孩子。JOIOJI桑想让自己孩子的名字和自己一样由“J、O、I”三个字母构成，并且想让“J、O、I”三个字母的出现次数恰好相同。

　　JOIOJI桑家有一份祖传的卷轴，上面写着一首长诗，长度为N，由“J、O、I”三个字母组成。JOIOJIさん想用诗中最长的满足要求的连续子串作为孩子的名字。

　　现在JOIOJI桑将这首长诗交给了你，请你求出诗中最长的、包含同样数目的“J、O、I”三个字母的连续子串。

Input

　　第一行一个正整数N，代表这首长诗的长度

　　接下来一行一个长度为N的字符串S，表示这首长诗，保证每个字符都是“J、O、I”三个字母中的一个

Output

　　输出一行一个正整数，代表最长的包含等数量“J、O、I”三个字母的最长连续子串的长度。如果不存在这样的子串，输出0

Sample Input

　　10

　　JOIIJOJOOI

Sample Output

　　6

HINT

　　选择“IIJOJO”这个子串，长度为6，包含“J、O、I”三个字母各2个，这是最长的满足要求的子串。

　　1<=N<=2*10^5

Solution

　　

　　假装这是个DP吧qwq。

Code

#include<iostream>    
#include<string>    
#include<algorithm>    
#include<cstdio>    
#include<cstring>    
#include<cstdlib>
#include<cmath>
#include<map>
using namespace std;  
typedef long long s64;
  
const int ONE = 2e5+5;

map <pair<int, int>, int> Map;
int n;
char S[ONE];
int f[ONE][4];
int Ans;

int get()
{    
        int res=1,Q=1;char c;    
        while( (c=getchar())<48 || c>57 ) 
        if(c=='-')Q=-1; 
        res=c-48;     
        while( (c=getchar())>=48 && c<=57 )    
        res=res*10+c-48;    
        return res*Q;
}

int id(char S) {return S=='J' ? 1:(S=='O' ? 2 : 3);}
int main()
{
        n = get();
        scanf("%s", S+1);
        Map[make_pair(0,0)] = 0;
        for(int i=1; i<=n; i++)
        {
            memcpy(f[i], f[i-1], sizeof(f[i-1]));
            f[i][id(S[i])]++;
            
            pair<int,int> t = make_pair(f[i][1] - f[i][2], f[i][2] - f[i][3]);
            if(Map.find(t) == Map.end()) Map[t] = i;
            else Ans = max(Ans, i - Map[t]);
        }
        printf("%d", Ans);
}