博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
P5161 WD与数列(后缀自动机+线段树合并)
阅读量:4881 次
发布时间:2019-06-11

本文共 3117 字,大约阅读时间需要 10 分钟。

没想出来→_→

首先不难看出要差分之后计算不相交也不相邻的相等子串对数,于是差分之后建SAM,在parent树上用线段树合并维护endpos集合,然后用启发式合并维护一个节点对另一个节点的贡献,于是总的时间复杂度为\(O(n\log^2n)\)

//minamoto#include
#define R register#define ll long long#define IT vector
::iterator#define fp(i,a,b) for(R int i=a,I=b+1;i
I;--i)#define go(u) for(int i=head[u],v=e[i].v;i;i=e[i].nx,v=e[i].v)#define gg(u) for(IT it=f[u].begin();it!=f[u].end();++it)using namespace std;char buf[1<<21],*p1=buf,*p2=buf;inline char getc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;}int read(){ R int res,f=1;R char ch; while((ch=getc())>'9'||ch<'0')(ch=='-')&&(f=-1); for(res=ch-'0';(ch=getc())>='0'&&ch<='9';res=res*10+ch-'0'); return res*f;}const int N=6e5+5;struct eg{int v,nx;}e[N];int head[N],tot;inline void add(R int u,R int v){e[++tot]={v,head[u]},head[u]=tot;}int fa[N],l[N],a[N],rt[N];map
ch[N];struct node{int s,ls,rs;ll xs;}t[N<<5];vector
*f[N],tmp[N];int n,m,ctot,cnt,las;ll ans;void ins(int c){ int p=las,np=++cnt;las=np,l[np]=l[p]+1; for(;p&&!ch[p].count(c);p=fa[p])ch[p][c]=np; if(!p)fa[np]=1; else{ int q=ch[p][c]; if(l[q]==l[p]+1)fa[np]=q; else{ int nq=++cnt;l[nq]=l[p]+1; ch[nq]=ch[q],fa[nq]=fa[q],fa[np]=fa[q]=nq; for(;p&&ch[p][c]==q;p=fa[p])ch[p][c]=nq; } }}void update(int &p,int l,int r,int x){ if(!p)p=++ctot;++t[p].s,t[p].xs+=x; if(l==r)return; int mid=(l+r)>>1; x<=mid?update(t[p].ls,l,mid,x):update(t[p].rs,mid+1,r,x);}int qaqs(int p,int l,int r,int ql,int qr){ if(!p)return 0;if(ql<=l&&qr>=r)return t[p].s; int mid=(l+r)>>1,res=0; if(ql<=mid)res+=qaqs(t[p].ls,l,mid,ql,qr); if(qr>mid)res+=qaqs(t[p].rs,mid+1,r,ql,qr); return res;}inline int querys(R int p,R int l,R int r){return (l>r||r<1||l>n)?0:qaqs(p,1,n,l,r);}ll qaqxs(int p,int l,int r,int ql,int qr){ if(!p)return 0;if(ql<=l&&qr>=r)return t[p].xs; int mid=(l+r)>>1;ll res=0; if(ql<=mid)res+=qaqxs(t[p].ls,l,mid,ql,qr); if(qr>mid)res+=qaqxs(t[p].rs,mid+1,r,ql,qr); return res;}inline ll queryxs(R int p,R int l,R int r){return (l>r||r<1||l>n)?0:qaqxs(p,1,n,l,r);}int merge(int x,int y){ if(!x||!y)return x|y; t[x].s+=t[y].s,t[x].xs+=t[y].xs; t[x].ls=merge(t[x].ls,t[y].ls),t[x].rs=merge(t[x].rs,t[y].rs); return x;}void dfs(int u){ int k=l[u]; go(u){ dfs(v);if(f[u]->size()
size())swap(f[u],f[v]),swap(rt[u],rt[v]); gg(v){ int x=*it; ll A=1ll*querys(rt[u],x-k-1,x-2)*(x-1)-queryxs(rt[u],x-k-1,x-2)+1ll*querys(rt[u],1,x-k-2)*k; ll B=1ll*queryxs(rt[u],x+2,x+k+1)-1ll*querys(rt[u],x+2,x+k+1)*(x+1)+1ll*querys(rt[u],x+k+2,n)*k; ans+=A+B,f[u]->push_back(x); }rt[u]=merge(rt[u],rt[v]); }}int main(){// freopen("testdata.in","r",stdin); n=read(); fp(i,1,n)a[i]=read(); fp(i,1,n-1)a[i]=a[i+1]-a[i],f[i]=&tmp[i]; ans=1ll*n*(n-1)>>1; --n,las=cnt=1; fp(i,1,n)ins(a[i]),f[las].push_back(i),update(rt[las],1,n,i); fp(i,2,cnt)add(fa[i],i); dfs(1);printf("%lld\n",ans); return 0;}

转载于:https://www.cnblogs.com/bztMinamoto/p/10207244.html

你可能感兴趣的文章
定位页面元素的位置
查看>>
关于IAsyncResult接口的CompletedSynchronously属性
查看>>
Python:一篇文章掌握Numpy的基本用法
查看>>
序列化与ArrayList 的elementData的修饰关键字transient
查看>>
学习进度17
查看>>
编译原理——算符优先分析文法(附源代码)
查看>>
jboss的启动过程
查看>>
渲染部分
查看>>
力扣——所有可能的路径
查看>>
关于VS项目平台的x86,x64,Any CPU以及Debug和Release的区别
查看>>
解密module_init幕后的故事
查看>>
9个移动网站优化的最佳实践
查看>>
李昌镐:苍老的青春(转载) 韩国围棋职业棋手
查看>>
JPA 使用报Named query not found错误
查看>>
FTP命令使用详解
查看>>
walmart weekly sales
查看>>
面试题07_用两个栈实现队列——剑指offer系列
查看>>
cocos2d-x3.2中加入Android手机震动
查看>>
css3处理sprite背景图压缩来解决H5网页在手机浏览器下图标模糊的问题
查看>>
温故而知新练习3
查看>>