bzoj 3238: [Ahoi2013]差异
Description
Input
一行,一个字符串S
Output
一行,一个整数,表示所求值
Sample Input
cacao
Sample Output
54
HINT
2<=N<=500000,S由小写英文字母组成
Source
好吧,看来品酒大会是蒯了这个题的套路。。。今天做了三个套路一模一样的题目。。。
看到这个题直接枚举lcp的长度L,然后就又是品酒大会并查集从大到小合并的套路了。。。
每个height相当于是要维护lcp为L的对数,以及len的和,合并的时候瞎jb乘法原理一下。。。
哎,思维越来越僵化了。。。话说没想到nlogn^2能跑过50w。。。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 | // MADE BY QT666 #include<cstdio> #include<algorithm> #include<cmath> #include<iostream> #include<cstring> using namespace std; typedef long long ll; const int N=600050; int gi(){ int x=0,flag=1; char ch=getchar(); while (ch< '0' ||ch> '9' ){ if (ch== '-' ) flag=-1;ch=getchar();} while (ch>= '0' &&ch<= '9' ) x=x*10+ch- '0' ,ch=getchar(); return x*flag; } int sa[N],len,y[N],rk,rnk[N],height[N],fa[N]; char a[N]; ll ansz[N],ansl[N],Ans,sz[N],le[N]; struct data{ int fir,sec,id; }x[N]; bool cmp( const data &a, const data &b){ if (a.fir==b.fir) return a.sec<b.sec; else return a.fir<b.fir; } void work2(){ rk=1;y[x[1].id]=rk; for ( int i=2;i<=len;i++){ if (x[i-1].fir!=x[i].fir||x[i-1].sec!=x[i].sec) rk++; y[x[i].id]=rk; } } void work(){ sort(x+1,x+1+len,cmp);work2(); for ( int i=1;i<=len;i<<=1){ for ( int j=1;j+i<=len;j++) x[j].fir=y[j],x[j].sec=y[j+i],x[j].id=j; for ( int j=len-i+1;j<=len;j++) x[j].fir=y[j],x[j].sec=0,x[j].id=j; sort(x+1,x+1+len,cmp);work2(); if (rk==len) break ; } } void get_height(){ int kk=0; for ( int i=1;i<=len;i++) rnk[sa[i]]=i; for ( int i=1;i<=len;i++){ if (kk) kk--; int j=sa[rnk[i]-1]; while (a[i+kk]==a[j+kk]) kk++; height[rnk[i]]=kk; } } struct Data{ int l,r,hi; }g[N]; bool cmp2( const Data &a, const Data &b){ return a.hi>b.hi; } int find( int x) { if (x!=fa[x]) fa[x]=find(fa[x]); return fa[x]; } void merge( int x, int y, int hi){ ansz[hi]+=sz[x]*sz[y]; ansl[hi]+=le[x]*sz[y]+le[y]*sz[x]; fa[x]=y;sz[y]+=sz[x];le[y]+=le[x]; } int main(){ scanf( "%s" ,a+1);len=strlen(a+1); for ( int i=1;i<=len;i++) x[i].id=i,x[i].fir=x[i].sec=a[i]- 'a' +1; work(); for ( int i=1;i<=len;i++) sa[y[i]]=i; for ( int i=1;i<=len;i++) sz[i]=1,fa[i]=i,le[i]=len-i+1; get_height(); for ( int i=2;i<=len;i++) g[i-1]=(Data){sa[i-1],sa[i],height[i]}; sort(g+1,g+len,cmp2); for ( int i=1;i<=len;i++){ int x=find(g[i].l),y=find(g[i].r); if (x!=y) merge(x,y,g[i].hi); } for ( int i=0;i<=len;i++) Ans+=ansl[i]-2*ansz[i]*i; printf( "%lld\n" ,Ans); return 0; } |
标签:
后缀数组
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 智能桌面机器人:用.NET IoT库控制舵机并多方法播放表情
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 手把手教你在本地部署DeepSeek R1,搭建web-ui ,建议收藏!
· 新年开篇:在本地部署DeepSeek大模型实现联网增强的AI应用
· Janus Pro:DeepSeek 开源革新,多模态 AI 的未来
· 互联网不景气了那就玩玩嵌入式吧,用纯.NET开发并制作一个智能桌面机器人(三):用.NET IoT库
· 【非技术】说说2024年我都干了些啥