BZOJ 4236 "JOIOJI"（前缀和+map+pair）

传送门：

　　[1]:BZOJ

　　[2]:洛谷

•题解

　　定义数组 a,b,c 分别表示 'J' , 'O' , 'I' 的前缀和；

　　要想使区间 (L,R] 满足条件当且仅当 a[R]-a[L] = b[R]-b[L] = c[R]-c[L];

　　那么，由 a[R]-a[L] = b[R]-b[L] ⇔ a[R]-b[R] = a[L]-b[L];

　　同理，由 b[R]-b[L] = c[R]-c[L] ⇔ b[R]-c[R] = b[L]-c[L];

　　提前预处理出 a,b,c 数组后；

　　对于 i 位置，查找之前是否含有满足 a_i-b_i = a_j-b_j && b_i-c_i = b_j-c_j 的位置 j，并且 j 尽可能的小；

　　如何高效的查找呢？

　　使用 map<pair<int ,int > , int >；

　　对于之前处理过的位置 j ，将 a_j-b_j 和 b_j-c_j 存入到 pair<int ,int > 中，每次查找是否存在 (a_i-b_i,b_i-c_i) 即可；

　　如果存在，求解当前答案，反之，将其加入到map中；

•Code

#include<bits/stdc++.h>
using namespace std;
#define pii pair<int ,int >
const int maxn=2e5+50;

int n;
char s[maxn];
int a[maxn];
int b[maxn];
int c[maxn];
map<pii ,int >f;

int Solve()
{
    f.clear();
    f[pii(0,0)]=0;///将(0,0)加入到f中
    a[0]=b[0]=c[0]=0;

    int len=strlen(s+1);
    for(int i=1;i <= len;++i)
    {
        a[i]=a[i-1]+(s[i] == 'J');
        b[i]=b[i-1]+(s[i] == 'O');
        c[i]=c[i-1]+(s[i] == 'I');
    }

    int ans=0;
    for(int i=1;i <= len;++i)
    {
        pii tmp=pii(a[i]-b[i],b[i]-c[i]);
        if(f.count(tmp))
            ans=max(ans,i-f[tmp]);
        else
            f[tmp]=i;
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    scanf("%s",s+1);

    printf("%d
",Solve());

    return 0;
}