码迷,mamicode.com
首页 > 其他好文 > 详细

[HihoCoder1413]Rikka with String

时间:2018-04-12 00:23:21      阅读:146      评论:0      收藏:0      [点我收藏+]

标签:ble   min   端点   长度   scanf   多少   algorithm   有一个   namespace   

vjudge

题意

给你一个串,问你把每个位置的字符替换成#后串中有多少本质不同的子串。
\(n\le 3*10^5\)

sol

首先可以计算出原串里面有多少本质不同的子串。显然就是\(\sum_{i=1}^{tot}len_i-len_{fa_i}\)

然后考虑把\(i\)位置替换成了#,那么你从\(1..i\)中任选左端点,\(i..n\)中任选右端点,这样的子串一定会是本质不同的。(一定含有字符#,满足长度不同或#的出现位置不同)

所以对于位置\(i\),答案先加上\(i*(n-i+1)\)

然后也可以再给答案加上原有的本质不同的子串数目,这样我们就只需要求出:把\(i\)位置改成#后有多少本质不同的子串不再出现了(被砍断)。

考虑一个\(SAM\)中的状态\(i\),设其最大长度\(len_i\)\(right\)集合中的最小最大元素分别为\(l_i,r_i\)

对于区间\([r_i-len_i+1,\min(r_i-len_{fa_i},l_i)]\)(如果合法的话),我们会给这个区间加上一个首项是\(1\)公差也是\(1\)的等差数列。

因为这个状态\(i\)本身是表示了\(len_i-len_{fa_i}+1\)个连续长度的子串,所以在这个区间内每向右移动一下就会有一个子串被砍断,因此是一个公差为\(1\)的等差数列。

对于区间\([min(r_i-len_{fa_i},l_i)+1,l_i]\)(如果合法的话),一旦#出现在这个区间里那么\(i\)这个状态表示的所有长度大于等于一定值的子串就都会被砍断。所以给这个区间集体加上前一个区间的长度,即\(\min(r_i-len_{fa_i},l_i)-(r_i-len_i+1)+1\)

等差数列用二阶差分维护一下就可以了,注意原一阶差分(第二种情况就是一个一阶差分)对应的二阶差分。所以构出\(SAM\)后的复杂度可以做到线性。

code

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 6e5+5;
int n,last=1,tot=1,tr[N][26],fa[N],len[N],l[N],r[N],t[N],a[N];
char s[N];long long sum,ans[N];
void extend(int c)
{
    int v=last,u=++tot;last=u;
    len[u]=len[v]+1;
    while (v&&!tr[v][c]) tr[v][c]=u,v=fa[v];
    if (!v) fa[u]=1;
    else{
        int x=tr[v][c];
        if (len[x]==len[v]+1) fa[u]=x;
        else{
            int y=++tot;
            memcpy(tr[y],tr[x],sizeof(tr[y]));
            fa[y]=fa[x];fa[x]=fa[u]=y;len[y]=len[v]+1;
            while (v&&tr[v][c]==x) tr[v][c]=y,v=fa[v];
        }
    }
}
int main()
{
    scanf("%d",&n);scanf("%s",s+1);
    memset(l,63,sizeof(l));
    for (int i=1;i<=n;++i) extend(s[i]-'a'),l[last]=r[last]=i;
    for (int i=1;i<=tot;++i) ++t[len[i]];
    for (int i=1;i<=tot;++i) t[i]+=t[i-1];
    for (int i=1;i<=tot;++i) a[t[len[i]]--]=i;
    for (int i=tot;i;--i)
    {
        l[fa[a[i]]]=min(l[fa[a[i]]],l[a[i]]);
        r[fa[a[i]]]=max(r[fa[a[i]]],r[a[i]]);
        sum+=len[i]-len[fa[i]];
    }
    for (int i=2;i<=tot;++i)
    {
        int L=r[i]-len[i]+1,R=min(r[i]-len[fa[i]],l[i]),Len=R-L+1;
        if (L<=R) ans[L]+=1,ans[R+1]-=Len+1,ans[R+2]+=Len;
        L=R+1;R=l[i];
        if (L<=R) ans[L]+=Len,ans[L+1]-=Len,ans[R+1]-=Len,ans[R+2]+=Len;
    }
    for (int i=1;i<=n;++i) ans[i]+=ans[i-1];
    for (int i=1;i<=n;++i) ans[i]+=ans[i-1];
    for (int i=1;i<=n;++i) printf("%lld ",1ll*i*(n-i+1)+sum-ans[i]);
    puts("");return 0;
}

[HihoCoder1413]Rikka with String

标签:ble   min   端点   长度   scanf   多少   algorithm   有一个   namespace   

原文地址:https://www.cnblogs.com/zhoushuyu/p/8799177.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!