BZOJ1921: [Ctsc2010]珠宝商（点分治+SAM）

传送门

题解：
点分治，如果点数≥n\ge \sqrt{n}≥n，则结合后缀树O(sze+m)O(sze+m)O(sze+m)处理出每个位置的开始，结束点并统计答案。否则从每个点开始暴力扩展路径。

时间复杂度O(nn)O(n \sqrt{n})O(nn)。

#include <bits/stdc++.h>
using namespace std;
typedef pair <int,int> pii;
typedef long long LL;inline int rd() {char ch=getchar(); int i=0,f=1;while(!isdigit(ch)) {if(ch=='-') f=-1; ch=getchar();}while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=getchar();}return i*f;
}const int N=1e5+50, B=150;
int n,m,mx,total,G,fa[N];
int sze[N],c1[N],c2[N],vis[N];
char s[N];
long long ans;
vector <int> edge[N];struct sam {char ch[N];int last,tot;int pos[N],son[N][26],tr[N][26],fail[N],len[N];int c[N],q[N];vector <int> edge[N];sam() {last=tot=1;}inline void extend(int c,int op) {int p=++tot; len[p]=len[last]+1; pos[p]=op;for(;last && !son[last][c];last=fail[last]) son[last][c]=p;if(!last) fail[p]=1;else {int q=son[last][c];if(len[q]==len[last]+1) fail[p]=q;else {int np=++tot; len[np]=len[last]+1; pos[np]=pos[q];memcpy(son[np],son[q],sizeof(son[np]));fail[np]=fail[q]; fail[q]=fail[p]=np;for(;last && son[last][c]==q;last=fail[last]) son[last][c]=np;}} last=p;}inline void init() {for(int i=1;i<=m;i++) extend(ch[i]-'a',i);for(int i=1;i<=tot;i++) if(fail[i]) edge[fail[i]].push_back(i);for(int i=1;i<=tot;i++) ++c[len[i]];for(int i=1;i<=tot;i++) c[i]+=c[i-1];for(int i=tot;i>=1;i--) q[c[len[i]]--]=i;memset(c,0,sizeof(c));for(int i=tot;i>=1;i--) {int u=q[i];if(pos[u]==len[u]) ++c[u];if(fail[u]) c[fail[u]]+=c[u];}for(int i=1;i<=tot;i++) {int u=q[i];for(int e=0,v;e<edge[u].size() && (v=edge[u][e]);++e)tr[u][ch[pos[v]-len[u]]-'a']=v;}memset(q,0,sizeof(q));}inline pii trans(pii p,char c) {if(!p.first) return pii(0,0);else if(p.second==len[p.first]) return pii(tr[p.first][c-'a'],p.second+1);else {if(ch[pos[p.first]-p.second]==c) return pii(p.first,p.second+1);else return pii(0,0);}}inline int trans2(int p,char c) {return son[p][c-'a'];}inline void add(pii p) {++q[p.first];}
} ori,rev;inline void calcG(int x,int f) {sze[x]=1;  int mx_son=0;for(int e=0,v;e<edge[x].size() && (v=edge[x][e]);++e)if(!vis[v] && v!=f) {calcG(v,x);sze[x]+=sze[v];mx_son=max(mx_son,sze[v]);}mx_son=max(mx_son,total-sze[x]);if(mx_son<=mx) mx=mx_son, G=x;
}inline void dfs(int x,int f,pii p1,pii p2) {p1=ori.trans(p1,s[x]); p2=rev.trans(p2,s[x]);ori.add(p1); rev.add(p2); sze[x]=1;for(int e=0,v;e<edge[x].size() && (v=edge[x][e]);++e)    if(!vis[v] && v!=f) {dfs(v,x,p1,p2);       sze[x]+=sze[v];}
}
inline void dfs(sam &t,int x,int sum,int flag) {sum+=t.q[x];if(t.pos[x]==t.len[x]) (flag ? c2[m-t.pos[x]+1] : c1[t.pos[x]])+=sum;;for(int e=0,v;e<t.edge[x].size() && (v=t.edge[x][e]);++e) dfs(t,v,sum,flag);
}int rt;
inline void dfs3(int x,int f,int p,int flag) {if(!p) return;ans+=flag*ori.c[p];for(int e=0,v;e<edge[x].size() && (v=edge[x][e]);++e) if(!vis[v] && v!=f) dfs3(v,x,ori.trans2(p,s[v]),flag);
}
inline void dfs2(int x,int f,int flag) {int p=1;if(flag<0) {for(int i=x;;i=fa[i]) {p=ori.trans2(p,s[i]); if(i==rt) break;}p=ori.trans2(p,s[G]); } if(flag<0) dfs3(rt,0,ori.trans2(p,s[rt]),flag);else dfs3(x,0,ori.trans2(p,s[x]),flag);for(int e=0,v;e<edge[x].size() && (v=edge[x][e]);++e) if(!vis[v] && v!=f) dfs2(v,x,flag);
}
inline void force_solve(int x,int flag) {rt=x; dfs2(x,0,flag);
}
inline void calc1(int t) {for(int i=1;i<=m;i++) c1[i]=c2[i]=0;dfs(ori,1,0,0); dfs(rev,1,0,1);for(int i=1;i<=m;i++) ans+=(LL)c1[i]*c2[i]*t;for(int i=1;i<=ori.tot;i++) ori.q[i]=0;for(int i=1;i<=rev.tot;i++) rev.q[i]=0;
}
inline void calc2(int x) {if(sze[x]<=B) {force_solve(x,-1);} else {pii p(1,0);dfs(x,G,ori.trans(p,s[G]),rev.trans(p,s[G]));calc1(-1);}
}
inline void dfs_pre(int x,int f) {fa[x]=f; for(int e=0,v;e<edge[x].size() && (v=edge[x][e]);++e) if(!vis[v] && v!=f) dfs_pre(v,x);}
inline void solve(int x) {if(total<=B) force_solve(x,1);else {pii p(1,0); vis[x]=1;ori.add(ori.trans(p,s[x]));rev.add(rev.trans(p,s[x]));for(int e=0,v;e<edge[x].size() && (v=edge[x][e]);++e)if(!vis[v]) dfs_pre(v,x), dfs(v,x,ori.trans(p,s[x]),rev.trans(p,s[x]));calc1(1);for(int e=0,v;e<edge[x].size() && (v=edge[x][e]);++e)if(!vis[v]) calc2(v);for(int e=0,v;e<edge[x].size() && (v=edge[x][e]);++e) if(!vis[v]) {mx=total=sze[v];calcG(v,x);solve(G);}}
}
int main() {n=rd(), m=rd();for(int i=1;i<n;i++) {int x=rd(), y=rd();edge[x].push_back(y);edge[y].push_back(x);} scanf("%s",s+1);scanf("%s",ori.ch+1);reverse_copy(ori.ch+1,ori.ch+m+1,rev.ch+1);ori.init(); rev.init();mx=total=n; calcG(1,0);solve(G);cout<<ans<<'\n';
}

BZOJ1921: [Ctsc2010]珠宝商（点分治+SAM）相关推荐

bzoj1921 [CTSC2010]珠宝商 SAM+后缀树+点分治
Description 有一棵n个节点的树和一个长度为m的字符串S,树上每个节点有一个字符.问对于任意的有序数对(x,y),从x到y路径组成的字符串在S中出现次数的和. n,m≤5⋅104n,m\le ...
BZOJ 1921: [Ctsc2010]珠宝商点分治套SAM
题目链接首先可以发现两种算法一. 暴力处理对"特征串"建SAM 枚举路径的一个端点,\(dfs\)另一个端点,同时维护在SAM上的位置. 每到一个位置会有SAM上对应节点的r ...
[BZOJ1921] [CTSC2010]珠宝商
Description Input 第一行包含两个整数 N,M,表示城市个数及特征项链的长度. 接下来的N-1 行, 每行两个整数 x,y, 表示城市 x 与城市 y 有直接道路相连.城市由1~N进行 ...
bzoj1921: [Ctsc2010]珠宝商
暴毙选手又被zo老师D费了暴力是n^2的都会有另一个点分的做法,就是看成两条链,然后在后缀树上跑,找到对应原串位置拼起来,是n*(logn+m) 然后就根号分治,树的大小超过阈值就点分,小于就暴力 ...
[CTSC2010]珠宝商 SAM+后缀树+点分治
[CTSC2010]珠宝商不错的题目看似无法做,n<=5e4,8s,根号算法? 暴力一: n^2,+SAM上找匹配点的right集合sz,失配了直接退出暴力二: O(m) 统计过lca=x ...
[CTSC2010]珠宝商（点分治+根号分治+后缀自动机）
[CTSC2010]珠宝商洛谷题目传送门简要题意给定一颗nnn个节点的树,和一个长度为mmm的模式串SSS 树上每个节点都有一个字符求树上所有路径的点的字符拼成的字符串在SSS中的出现次数之和 ...
bzoj 1921: [Ctsc2010]珠宝商
题意: 给一棵树,每个点上有一个字母,问对于所有(x,y),求x到y的路径所组成的字符串在S中出现次数的和. 题解: 先上题解:begined CTSC2010 珠宝商新解然后说说个人的垃圾理解. ...
P4218 [CTSC2010]珠宝商
P4218 [CTSC2010]珠宝商神题... 可以想到点分治,细节不写了... (学了个新姿势,sam可以在前面加字符但是一次点分治只能做到\(O(m)\),考虑\(\sqrt n\)点分治, ...
bzoj1921 CTSC2010 jewelry
大概是四节课写代码,三节课debug,七个长相差不多的dfs穿插其中-- 好了不吐槽了: 题意: 给定一棵点上有字母的树,和一个母串求树上点两两之间 N2N^2 条路径在母串中匹配次数和这里采用的 ...

BZOJ1921: [Ctsc2010]珠宝商（点分治+SAM）

BZOJ1921: [Ctsc2010]珠宝商（点分治+SAM）相关推荐

最新文章

热门文章