BZOJ-3238: [Ahoi2013]差异(Suffix Array + Cartesian Tree)

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=3238

这是我ORZ了网上的题解才知道的555:

首先做一次后缀数组,求出sa[],height[],然后对于height[2..len(s)]建立Cartesian Tree,那么每个节点对于题目中项lcp()的贡献为(size[left[v]]+1)*(size[right[v]]+1)(很好证明。。。略),然后得出答案。

代码:

速度依旧被虐成渣。。。

b219ebc4b74543a9194daec31c178a82b901146e.jpg.png
#include <cstdio>
#include <algorithm>
#include <cstring>
 
using namespace std;
 
#define MAXN 500010
#define ll long long 
#define F(t) father[t]
#define L(t) left[t]
#define R(t) right[t]
#define K(t) key[t]
#define S(t) size[t]
#define clear(t) memset(t,0,sizeof(t))
 
int getstring(char *s) {
    int ch,len=0;
    for (ch=getchar();!(ch>='a'&&ch<='z');ch=getchar()) ;
    s[++len]=ch;
    for (ch=getchar();ch>='a'&&ch<='z';ch=getchar()) s[++len]=ch;
    return len;
}
 
char s[MAXN];
int n,height[MAXN],sa[MAXN],rank[MAXN],w[MAXN],r[MAXN],x[MAXN],y[MAXN];
 
void make_sa() {
    int N,b=1,M=n;
    s[sa[0]=0]='$';
    for (int i=0;i++<n;) M=max(M,rank[i]=s[i]);
    do {
        for (int i=0;i++<n;) x[i]=rank[i],y[i]=i+b<=n?rank[i+b]:0;
        b<<=1;
        for (int i=0;i<=M;i++) w[i]=0;
        for (int i=0;i++<n;) w[y[i]]++;
        for (int i=0;i++<M;) w[i]+=w[i-1];
        for (int i=0;i++<n;) r[w[y[i]]--]=i;
        for (int i=0;i<=M;i++) w[i]=0;
        for (int i=0;i++<n;) w[x[r[i]]]++;
        for (int i=0;i++<M;) w[i]+=w[i-1];
        for (int i=n;i;i--) sa[w[x[r[i]]]--]=r[i];
        N=0;
        for (int i=0;i++<n;) {
            if (i==1||x[sa[i]]!=x[sa[i-1]]||y[sa[i]]!=y[sa[i-1]]) N++;
            rank[sa[i]]=N;
        }
    } while (N<n);
    int k=0;
    for (int i=0;i++<n;) {
        height[rank[i]]=k;
        for (int j=k;i+j<=n&&sa[rank[i]-1]+j<=n&&s[i+j]==s[sa[rank[i]-1]+j];j++) height[rank[i]]++;
        k=max(height[rank[i]]-1,0);
    }
}
 
int father[MAXN],left[MAXN],right[MAXN],roof=0;
ll size[MAXN],key[MAXN];
 
void dfs(int t) {
    S(t)=1;
    if (L(t)) dfs(L(t)),S(t)+=S(L(t));
    if (R(t)) dfs(R(t)),S(t)+=S(R(t));
}
 
ll Solve() {
    ll rec=0;
    for (int i=0;i++<n;) {
        ll ret=n-i+1;
        rec+=ret;
    }
    rec*=(n-1);
    F(0)=L(0)=R(0)=K(0)=S(0)=0;
    int t=0;
    for (int i=2;i<=n;i++) {
        K(i)=height[i];
        if (!roof) { roof=t=i; continue; }
        for (;t&&K(t)>K(i);t=F(t)) ;
        if (t) {
            L(i)=R(t),R(t)=i,F(i)=t;
        } else if (K(roof)>K(i)) L(i)=roof,roof=i; else L(i)=R(roof),R(roof)=i,F(i)=roof;
        t=i;
    }
    dfs(roof);
    for (int i=2;i<=n;i++) {
        ll ret=(S(L(i))+1)*(S(R(i))+1)*K(i)*2;
        rec-=ret;
    }
    return rec;
}
 
int main() {
    n=getstring(s);
    make_sa();
    printf("%lld\n",Solve());
    return 0;
}

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,445评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,889评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,047评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,760评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,745评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,638评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,011评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,669评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,923评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,655评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,740评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,406评论 4 320
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,995评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,961评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,197评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,023评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,483评论 2 342

推荐阅读更多精彩内容

  • 超高速音视频编码器用法: ffmpeg [options] [[infile options] -i infile...
    吉凶以情迁阅读 4,582评论 0 4
  • 周六,我们举办了一个亲友联谊会,我过去充当一名工作人员。 还没起床,就听到外面滴答滴答的声音,我知道这是下雨了,真...
    现进阅读 795评论 0 0
  • 由于gdb的原因,最后还是失败了。。mark赶紧绕路。。。转到用xcode来弄了。。https://www.jia...
    FlyingPenguin阅读 1,470评论 0 0
  • 和阿虹成为好朋友,从此便不孤单。 与同学们也渐渐熟悉,心情好了许多。每天在宿舍的时光是最悠闲自由快活的。 我性格中...
    暖香阁主阅读 286评论 1 1
  • 陈瑛阅读 160评论 0 1