[bzoj4199][后缀数组][后缀自动机]品酒大会

4199: [Noi2015]品酒大会

Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 1391 Solved: 793
[Submit][Status][Discuss]
Description

一年一度的“幻影阁夏日品酒大会”隆重开幕了。大会包含品尝和趣味挑战两个环节，分别向优胜者颁发“首席品

酒家”和“首席猎手”两个奖项，吸引了众多品酒师参加。在大会的晚餐上，调酒师Rainbow调制了 n 杯鸡尾酒。
这 n 杯鸡尾酒排成一行，其中第 i 杯酒 (1≤i≤n) 被贴上了一个标签 s_i ，每个标签都是 26 个小写英文字母
之一。设 Str(l,r) 表示第 l 杯酒到第 r 杯酒的 r-l+1 个标签顺次连接构成的字符串。若 Str(p,po)=Str(q,qo
) ，其中 1≤p≤po≤n,1≤q≤qo≤n,p≠q,po-p+1=qo-q+1=r ，则称第 p 杯酒与第 q 杯酒是“ r 相似”的。当
然两杯“ r 相似”(r>1)的酒同时也是“ 1 相似”、“ 2 相似”、……、“ (r-1) 相似”的。在品尝环节上，
品酒师Freda轻松地评定了每一杯酒的美味度，凭借其专业的水准和经验成功夺取了“首席品酒家”的称号，其中
第 i 杯酒 (1≤i≤n) 的美味度为 a_i 。现在Rainbow公布了挑战环节的问题：本次大会调制的鸡尾酒有一个特点
，如果把第 p 杯酒与第 q 杯酒调兑在一起，将得到一杯美味度为 a_p a_q 的酒。现在请各位品酒师分别对于 r=
0,1,2,?,n-1 ，统计出有多少种方法可以选出 2 杯“ r 相似”的酒，并回答选择 2 杯“ r 相似”的酒调兑可以
得到的美味度的最大值。
Input

输入文件的第1行包含1个正整数 n ，表示鸡尾酒的杯数。
第 2 行包含一个长度为 n 的字符串 S ，其中第 i 个字符表示第 i 杯酒的标签。
第 3 行包含 n 个整数，相邻整数之间用单个空格隔开，其中第 i 个整数表示第 i 杯酒的美味度 a_i 。
n=300,000 |a_i |≤1,000,000,000
Output

输出文件包括 n 行。第 i 行输出 2 个整数，中间用单个空格隔开。

第 1 个整数表示选出两杯“ (i-1)” ” 相似”的酒的方案数，
第 2 个整数表示选出两杯“ (i-1) 相似”的酒调兑可以得到的最大美味度。
若不存在两杯“ (i-1) 相似”的酒，这两个数均为 0 。
Sample Input

10

ponoiiipoi

2 1 4 7 4 8 3 6 4 7
Sample Output

45 56

10 56

3 32

0 0

0 0

0 0

0 0

0 0

0 0

0 0

【样例说明1】

用二元组 (p,q) 表示第 p 杯酒与第 q 杯酒。

0 相似：所有 45 对二元组都是 0 相似的，美味度最大的是 8×7=56 。

1 相似： (1,8) (2,4) (2,9) (4,9) (5,6) (5,7) (5,10) (6,7) (6,10) (7,10) ，最大的 8×7=56 。

2 相似： (1,8) (4,9) (5,6) ，最大的 4×8=32 。

没有 3,4,5,?,9 相似的两杯酒，故均输出 0 。
HINT

Source

sa:

这题的a是有负数的。

感觉挺毒瘤的，边界情况有点多，注意不要重复计算。
考虑第一问很经典，单调栈求一下hei的影响区间即可。第二问的话其实就是合并一下左右区间。把它们的最大值相乘，最小值相乘。

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
using namespace std;
const int N=510000;
typedef long long LL;
int n;
int sa[N],rank[N],hei[N],w[N+1],x[N];
LL Max[N][25],Min[N][25];
char sr[N];

inline int read()
{
    char c;
    bool flag=false;
    while((c=getchar())>'9'||c<'0')
    if(c=='-')flag=true;
    int res=c-'0';
    while((c=getchar())>='0'&&c<='9')
    res=(res<<3)+(res<<1)+c-'0';
    return flag?-res:res;
}

inline void Sa()
{
    int u,v,m,p;
    m=255;
    for(int i=1;i<=n;++i) w[x[i]=sr[i]-'0']++;
    for(int i=1;i<=m;++i) w[i]+=w[i-1];
    for(int i=n;i>=1;--i) sa[w[x[i]]--]=i;
    for(int j=1;j<=n;j<<=1)
    {
        p=0;
        for(int i=n-j+1;i<=n;++i) rank[++p]=i;
        for(int i=1;i<=n;++i) if(sa[i]-j>0) rank[++p]=sa[i]-j;
        for(int i=1;i<=m;++i) w[i]=0;
        for(int i=1;i<=n;++i) w[x[i]]++;
        for(int i=1;i<=m;++i) w[i]+=w[i-1];
        for(int i=n;i>=1;--i)
            sa[w[x[rank[i]]]--]=rank[i];
        m=0;
        for(int i=1;i<=n;++i)
        {
            u=sa[i];v=sa[i-1];
            if(x[u]!=x[v]||x[u+j]!=x[v+j]) m++;
            rank[u]=m;
        }
        if(m==n) break;
        for(int i=1;i<=n;++i) swap(x[i],rank[i]);
    }
    int len=0;
    for(int i=1;i<=n;++i)
    {
        len=max(len-1,0);
        while(sr[i+len]==sr[sa[rank[i]-1]+len]) len++;
        hei[rank[i]]=len;
    }
}

int Q[25],a[N];
inline void Rmq()
{
    for(int i=1;i<=n;++i)
        Min[i][0]=Max[i][0]=a[sa[i]];
    Q[0]=1;
    for(int i=1;i<=19;++i)
        Q[i]=Q[i-1]<<1;
    for(int j=1;j<=19;++j)
    for(int i=1;i<=n;++i)
    {
        if(i+Q[j-1]>n)break;
        Max[i][j]=max(Max[i][j-1],Max[i+Q[j-1]][j-1]),
        Min[i][j]=min(Min[i][j-1],Min[i+Q[j-1]][j-1]);
    }
}
LL ma[5],mi[5];
inline LL query(int l,int r,int k)
{
    int i,len=r-l+1;
    for(i=19;i>=0;--i)
        if(Q[i]<=len) break;
    ma[k]=max(Max[l][i],Max[r-Q[i]+1][i]);
    mi[k]=min(Min[l][i],Min[r-Q[i]+1][i]);
}

const int inf=1e9;
int top;
int L[N],R[N],q[N];
LL ans1[N],ans2[N];
int main()
{
//  freopen("4199.in","r",stdin);
    n=read();
    scanf("%s",sr+1);
    for(int i=1;i<=n;++i)
        a[i]=read();
    Sa();
//  for(int i=1;i<=n;++i)
//  printf("%d\n",hei[i]);
    hei[n+1]=-1;
    top=0;
    for(int i=n;i>=1;--i)
    {
        while(top&&hei[q[top]]>=hei[i]) L[q[top--]]=i+1;
        q[++top]=i;
    }
    top=1;
    for(int i=1;i<=n+1;++i)
    {
        while(top&&hei[q[top]]> hei[i]) R[q[top--]]=i-1;
        q[++top]=i;
    }
    Rmq();
    LL l,r;
    for(int i=0;i<=n;++i)
    ans2[i]=-1e19;
    for(int i=2;i<=n;++i)
    {
        l=i-L[i]+1;
        r=R[i]-i+1;
        ans1[hei[i]]+=l*r;
        query(L[i]-1,i-1,0);query(i,R[i],1);
        ans2[hei[i]]=max(ans2[hei[i]],mi[0]*mi[1]);
        ans2[hei[i]]=max(ans2[hei[i]],ma[0]*ma[1]);
    }

    for(int i=n-1;i>=0;--i)
    ans1[i]+=ans1[i+1],ans2[i]=max(ans2[i],ans2[i+1]);
    for(int i=0;i<n;++i)
    if(!ans1[i]) printf("0 0\n");
    else
    printf("%lld %lld\n",ans1[i],ans2[i]);
}

sam:

第一问求得是某一种lcp的选取方案。不怎么好做，先反串吧。
变成了某一种公共后缀长度的选取方案。树d即可
考虑第二问。2个r相似的串的乘积最大值。那就是维护一下最大，最小乘一乘就好了。
为什么我这题做sa和sam全tm被边界卡一晚上，可以说是很骚了

#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cmath>
using namespace std;
typedef long long ll;

int n,m;

inline int read()
{
    char c;
    bool pd=0;
    while((c=getchar())>'9'||c<'0')
    if(c=='-') pd=1;
    int res=c-'0';
    while((c=getchar())>='0'&&c<='9')
    res=(res<<3)+(res<<1)+c-'0';
    return pd?-res:res;
}
const int N=610000;
int cnt=1,fa[N],Max[N],Min[N],go[N][26];
int last=1,in[N],size[N],val[N];
ll ans1[N],ans2[N];
inline void extend(int x)
{
    int p=last;
    int np=last=++cnt;
    val[np]=val[p]+1;
    while(!go[p][x]) go[p][x]=np,p=fa[p];
    if(!p) fa[np]=1,in[1]++;
    else
    {
        int q=go[p][x];
        if(val[q]==val[p]+1) fa[np]=q,in[q]++;
        else
        {
            int nq=++cnt;
            val[nq]=val[p]+1;
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            Max[nq]=-1e9-7;
            Min[nq]=1e9+7;
            for(int i=0;i<=25;++i) go[nq][i]=go[q][i];
            in[nq]=2;
            while(go[p][x]==q) go[p][x]=nq,p=fa[p];
        }
    }
}char sr[N];
int q[N],v[N];
int main()
{
//  freopen("4199.in","r",stdin);
//  freopen(".out","w",stdout);
    n=read();
    scanf("%s",sr+1);
    for(int i=1;i<=n;++i) v[i]=read();
    m=n>>1;
    for(int i=1;i<=m;++i) swap(sr[i],sr[n-i+1]),swap(v[i],v[n-i+1]);
    for(int i=1;i<=n;++i)
    {
        size[cnt+1]=1;
        Max[cnt+1]=Min[cnt+1]=v[i];
        extend(sr[i]-'a');
    }
    int t=0,w=0;
    for(int i=1;i<=cnt;++i)
    if(!in[i]) q[++w]=i;
    for(int i=1;i<=n;++i) ans2[i]=-1e18;
    while(t<w)
    {
        int u=q[++t];
        if(u==1) break;
        if(!(--in[fa[u]])) q[++w]=fa[u];
        ans1[val[fa[u]]]+=(ll)size[fa[u]]*size[u];
        size[fa[u]]+=size[u];
        if(Max[u]!=-1e9-7&&Max[fa[u]]!=-1e9-7)
        ans2[val[fa[u]]]=max(ans2[val[fa[u]]],(ll)Max[u]*Max[fa[u]]);
        if(Min[u]!=1e9+7&&Min[fa[u]]!=1e9+7)
        ans2[val[fa[u]]]=max(ans2[val[fa[u]]],(ll)Min[u]*Min[fa[u]]);
        Max[fa[u]]=max(Max[fa[u]],Max[u]);
        Min[fa[u]]=min(Min[fa[u]],Min[u]);
    }
    for(int i=n-1;i>=0;--i)
    ans1[i]+=ans1[i+1],ans2[i]=max(ans2[i],ans2[i+1]);
    for(int i=0;i<n;++i)
    if(ans1[i]) printf("%lld %lld\n",ans1[i],ans2[i]);
    else printf("0 0\n");
}