【GDOI2018Day2模拟4.18】奥利卡的诗

Description

Input

从文件poem.in中读入数据.
第一行两个整数n.
接下来若干行有n个单词，用空格或回车隔开.

Output

输出到文件poem.out中.
输出n行n个整数，第i个整数表示elegancy(i).

Sample Input

输入1：
2
nyg
nyx

Sample Output

输出1：
6
18

Data Constraint

n<=105,∑|s|<=3∗105 n

Solution

答案显然是每个子串的数量的平方和
子串数量，考虑SAM
对所有字符串建trie，然后建SAM
初始时所有点right集大小为0
用数据结构维护答案，因为right集的大小是在fail树上的，所以对fail树链剖然后用线段树维护平方和
注意一个状态中存储的字符串是[min,max]，而不是一个
所以方便一点的处理方法是每次修改时考虑统计1~r对r+1的贡献和r+1对1~r+1的贡献
具体看程序吧

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define N 1001000
#define ll long long
using namespace std;
int n,last[N],d[N],tr[N][26],tot=1,m,las[N],nxt[N],to[N],son[N],top[N],sz[N],fa[N],dfn[N],s[N],en[N];
ll ans=0;
struct{int fail,len,to[26];
}t[N];
struct{ll lz,v,s;
}g[N*3];
void putin(int x,int y)
{nxt[++tot]=las[x];las[x]=tot;to[tot]=y;
}
void add(int x,int las)
{int p=las,np=++tot;t[np].len=t[p].len+1;for(;p&&t[p].to[x]==0;p=t[p].fail) t[p].to[x]=np;if(p==0){t[np].fail=1;return;}int q=t[p].to[x];if(t[p].len+1==t[q].len) t[np].fail=q;else{int nq=++tot;t[nq]=t[q];t[nq].len=t[p].len+1;t[q].fail=t[np].fail=nq;for(;p&&t[p].to[x]==q;p=t[p].fail) t[p].to[x]=nq;}
}
void bfs()
{int he=0,ta=1;d[1]=1;while(he<ta){int x=d[++he];fo(i,0,25)if(tr[x][i]){d[++ta]=tr[x][i];last[tr[x][i]]=tot+1;add(i,last[x]);}}
}
void dg1(int x)
{sz[x]=1;for(int i=las[x];i;i=nxt[i]){int y=to[i];fa[y]=x;dg1(y);sz[x]+=sz[y];if(sz[y]>sz[son[x]]) son[x]=y;}
}
void dg2(int x)
{dfn[x]=++tot;if(son[x]) top[son[x]]=top[x],dg2(son[x]);for(int i=las[x];i;i=nxt[i]){int y=to[i];if(y==son[x]) continue;top[y]=y;dg2(y);}
}
void build(int v,int i,int j,int x,int y)
{if(i==j){g[v].v=t[y].len-t[t[y].fail].len;return;}int m=(i+j)/2;if(x<=m) build(v*2,i,m,x,y);else build(v*2+1,m+1,j,x,y);g[v].v=g[v*2].v+g[v*2+1].v;
}
void down(int v)
{if(g[v].lz==0) return;g[v*2].lz+=g[v].lz;g[v*2+1].lz+=g[v].lz;g[v*2].s+=g[v].lz*g[v*2].v;g[v*2+1].s+=g[v].lz*g[v*2+1].v;g[v].lz=0;
}
void ins(int v,int i,int j,int x,int y)
{if(i==x&&j==y){g[v].lz++;g[v].s=g[v].s+g[v].v;return;}int m=(i+j)/2;down(v);if(y<=m) ins(v+v,i,m,x,y);else if(x>m) ins(v+v+1,m+1,j,x,y);else ins(v+v,i,m,x,m),ins(v+v+1,m+1,j,m+1,y);g[v].s=g[v*2].s+g[v*2+1].s;
}
void get(int v,int i,int j,int x,int y)
{if(i==x&&j==y){ans+=g[v].s;return;}int m=(i+j)/2;down(v);if(y<=m) get(v+v,i,m,x,y);else if(x>m) get(v+v+1,m+1,j,x,y);else get(v+v,i,m,x,m),get(v+v+1,m+1,j,m+1,y);
}
void change(int x)
{x=last[x];int y=x;while(y>1){x=top[y];ins(1,1,m,dfn[x],dfn[y]);y=fa[x];}
}
void calc(int x)
{x=last[x];int y=x;while(y>1){x=top[y];get(1,1,m,dfn[x],dfn[y]);y=fa[x];}
}
int main()
{freopen("poem.in","r",stdin);freopen("poem.out","w",stdout);scanf("%d\n",&n);fo(i,1,n){int c=getchar()-97;int x=1;en[i]=en[i-1];for(;c>=0&&c<=26;c=getchar()-97){if(tr[x][c]==0) tr[x][c]=++tot;x=tr[x][c];s[++en[i]]=c;}}last[1]=tot=1;bfs();m=tot;tot=0;fo(i,2,m) putin(t[i].fail,i);tot=0;dg1(1);son[1]=0;dg2(1);fo(i,1,m)build(1,1,m,dfn[i],i);fo(i,1,n){int x=1;fo(j,en[i-1]+1,en[i]){x=tr[x][s[j]];calc(x);}x=1;fo(j,en[i-1]+1,en[i]){x=tr[x][s[j]];change(x);}x=1;fo(j,en[i-1]+1,en[i]){x=tr[x][s[j]];calc(x);}printf("%lld\n",ans);}
}

【GDOI2018Day2模拟4.18】奥利卡的诗相关推荐

[JZOJ5666]【GDOI2018Day2模拟4.18】法力风暴（分治NTT 模板）
Description 有 2≤n≤105,0≤Ai,k≤109 2 ≤ n ≤ 10 5 , 0 ≤ A i , k ≤ 10 9 2 ≤ n ≤ 10^5 ,0 ≤ A_i ,k ≤ 10^9 S ...
NOIP模拟测试18「引子·可爱宝贝精灵·相互再归的鹅妈妈」
待补引子题解大模拟,注意细节代码1 #include<bits/stdc++.h> using namespace std; int n,m;char a[1005][1005]; ...
异常练习（模拟不满18岁不可进入网吧）
代码如下: public class NetBar {private int age;public int getAge() {return age;}public void setAge(int a ...
JZOJ 5911. 【NOIP2018模拟10.18】Travel
题目 EZ常常在假期环游世界,他准备去N(N<=100000)个国家之多,一些国家有航线连接,由于EZ同学有一定的强迫症,任意两个国家之间都能通过航路直接或间接到达,并且这样的路径仅有一种.(简 ...
2019.8.12 NOIP模拟测试18 反思总结
写个博客总是符合要求的对吧回来以后第一次悄悄参加考试,昨天全程围观- 然后喜提爆炸120分wwwwwwwww T1用了全机房最慢的写法,导致改掉死循环T掉的一个点以后还是死活过不了最后一个点.T2全 ...
JZOJ 5691. 【GDOI2018Day2模拟4.25】求和
Description master对树上的求和非常感兴趣.他生成了一棵有根树,并且希望多次询问这棵树上一段路径上所有节点深度的k次方和,而且每次的k可能是不同的.此处节点深度的定义是这个节点到根的路 ...
JZOJ 5689. 【GDOI2018Day2模拟4.25】二进制
Description Pupil 发现对于一个十进制数,无论怎么将其的数字重新排列,均不影响其是不是3 的倍数.他想研究对于二进制,是否也有类似的性质.于是他生成了一个长为n的二进制串,希望你对于这 ...
JZOJ 5678. 【GDOI2018Day2模拟4.21】果树
Description NiroBC 姐姐是个活泼的少女,她十分喜欢爬树,而她家门口正好有一棵果树,正好满足了她爬树的需求. 这颗果树有N个节点,节点标号 1-N.每个节点长着一个果子,第i个节点上的 ...
JZOJ 5677. 【GDOI2018Day2模拟4.21】纽约
Description 印度洋暖流温润着纽约,四季丰沛的雨水造就了一望无际的大草原.蒙古包是纽约最独特的一道风景线,每至二月中旬,纽约的土著傣族人民又开始半年一度的转场了. 由于牲畜和行李过多,牧民 ...