UPC2019/07/12

Contest RankList – 2019年第二阶段我要变强个人训练赛第十四场

问题 A: JOIOJI

题目描述

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

最近，JOIOJI 桑喜当爹。JOIOJI 桑想让自己孩子的名字和自己一样由J,O,I三种字母构成，并且想让J,O,I三个字母的出现次数恰好相同。

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

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

输入

第一行一个正整数N，代表这首长诗的长度。
接下来一行一个长度为N的字符串S，表示这首长诗，保证每个字符都是J,O,I三个字母中的一个。

输出

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

样例输入

10
JOIIJOJOOI

样例输出

6

题意

找到最长的相同数量J，O，I序列

题解：

统计从头到某个位置一共有多少个J，O，I

x[i],y[i],z[i]表示前i个位置分别有多少个J,O,I

然后找到一队 x[j]-By[j]==x[i]-y[i] && y[j]-z[j]==y[j]-z[j] 就好了

化简即 x[j]-x[i]=y[j]-y[i],y[j]-y[i]=z[j]-z[i]

代码

// An highlighted block
#include<bits/stdc++.h> 
#include <map>
using namespace std;
long long read()
{
    long long x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

char s[200010];
map<pair<int,int> ,int> H;
int ans=0;
int x,y,z;
int main()
{
    int n=read();
    scanf("%s",s+1);
    H[make_pair(0,0)]=0;
    for(int i=1;i<=n;i++)
    {
        if(s[i]=='J')
            x++;
        if(s[i]=='O')
            y++;
        if(s[i]=='I')
            z++;
        if(H.find(make_pair(x-y,y-z))==H.end())
            H[make_pair(x-y,y-z)]=i;
        else
            ans = max(ans,i-H[make_pair(x-y,y-z)]);
    }
    cout<<ans;
    return 0;
}