bzoj 2213: [Poi2011]Difference 乱搞

题意

已知一个长度为n的由小写字母组成的字符串，求其中连续的一段，满足该段中出现最多的字母出现的个数减去该段中出现最少的字母出现的个数最大。求这个个数。
该段必须出现两个以上不同的字母。
1<=n<=1000000

分析

这题的话，自己yy出来了一个很棒的做法，然后对于如何处理最小出现次数不能为0的条件，借鉴了一下题解的方法。
我的方法就是把每一种字母出现的位置分别记录下来，然后每次枚举两个字母，设其分别为出现最多的字母和出现最少的字母。
然后我们可以把这两种字母所在的位置归并一下，那么问题就变成了有一个01序列，要求找连续的一段，使得1的个数减去0的个数最大且0的个数不能为0.
考虑将1的权值设为1,0的权值设为-1，那么就是要求最大子段和了。
如果没有0的个数不能为0这个限制的话，我们只用维护一个前缀和的最小值就好了。加上这个限制的话，我们可以维护一个最小值和次小值，使得这其对应位置的前缀0的个数不同，然后就可以直接做了。
显然每个位置只会被归并25次，最多被扫过50次，所以复杂度是O(50n)

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=1000005;

int n,s[N],next[N],last[30],cnt[N],a[N],a1,c[N],ans,tot[N];
char ch[N];

void work(int x,int y)
{
    int fir=0,sec=a1,s1=0,s2=-1;
    for (int i=1;i<=a1;i++)
    {
        cnt[i]=cnt[i-1]+(c[a[i]]==y);
        s[i]=s[i-1]+(c[a[i]]==x?1:-1);
        if (cnt[i]==s1) fir=min(fir,s[i]);
        else if (cnt[i]==s2) sec=min(sec,s[i]);
        else if (s[i]<fir) sec=fir,s2=s1,fir=s[i],s1=cnt[i];
        else if (s[i]<sec) sec=s[i],s2=cnt[i];
        if (cnt[i]-s1>0) ans=max(ans,s[i]-fir);
        else ans=max(ans,s[i]-sec);
    }
}

int main()
{
    scanf("%d",&n);
    scanf("%s",ch+1);
    for (int i=1;i<=n;i++) c[i]=ch[i]-'a',next[i]=last[c[i]],last[c[i]]=i;
    for (int i=0;i<25;i++)
        for (int j=i+1;j<26;j++)
        {
            if (!last[i]||!last[j]) continue;
            a1=0;
            int p=last[i],q=last[j];
            while (p||q)
                if (p>q) a[++a1]=p,p=next[p];
                else a[++a1]=q,q=next[q];
            work(i,j);
            work(j,i);
        }
    printf("%d",ans);
    return 0;
}