CF504E Misha and LCP on Tree(树链剖分+后缀树组)
1A真舒服。
喜闻乐见的树链剖分+SA。
一个初步的想法就是用树链剖分,把两个字符串求出然后hash+二分求lcp。。。不存在的。
因为考虑到这个字符串是有序的,我们需要把每一条重链对应的字符串和这个重链反过来对应的字符串拼起来构成一个新的字符串。我们用树链剖分拼出两个字符串。用树剖拼出的这两个字符串,一定是重链拼成的字符串上一个一个区间,我们记录这些区间的左右端点。然后我们就是要一个一个处理区间。求两个区间对应字符串的lcp,我们直接上SA+ST表就行,求出lcp之后就在这些区间上分情况讨论,然后就结束了。
具体看代码,太丑了也看不出来什么。
#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=601000;
int cnt,head[N];
int size[N],fa[N],dep[N],dfn[N],top[N],tot,son[N];
int s[N],id[N][2],cn,Top[N];
int sa[N],x[N],y[N],len,rk[N],height[N],m,mn[N][20],c[N];
int L[N][2],R[N][2],NUM[N],Num[N],n;
char S[N];
struct edge{
int to,nxt;
}e[N];
void add(int u,int v){
cnt++;
e[cnt].nxt=head[u];
e[cnt].to=v;
head[u]=cnt;
}
void dfs1(int u,int f){
size[u]=1;
fa[u]=f;
dep[u]=dep[f]+1;
int maxson=-1;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==f)continue;
dfs1(v,u);
if(size[v]>maxson){
maxson=size[v];
son[u]=v;
}
size[u]+=size[v];
}
}
void dfs2(int u,int tp){
dfn[u]=++tot;
top[u]=tp;
if(son[u])dfs2(son[u],tp);
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==fa[u]||v==son[u])continue;
Top[++cn]=v;
dfs2(v,v);
}
}
void build(int u){
s[++len]=S[u];
id[dfn[u]][0]=len;
if(son[u]==0){
id[dfn[u]][0]=len;
s[++len]=S[u];
id[dfn[u]][1]=len;
return;
}
build(son[u]);
s[++len]=S[u];
id[dfn[u]][1]=len;
}
void get_sa(){
for(int i=1;i<=len;i++)c[x[i]=s[i]]++;
for(int i=1;i<=m;i++)c[i]+=c[i-1];
for(int i=len;i>=1;i--)sa[c[x[i]]--]=i;
for(int k=1;k<=len;k<<=1){
int num=0;
for(int i=len-k+1;i<=len;i++)y[++num]=i;
for(int i=1;i<=len;i++)if(sa[i]>k)y[++num]=sa[i]-k;
for(int i=1;i<=m;i++)c[i]=0;
for(int i=1;i<=len;i++)c[x[i]]++;
for(int i=1;i<=m;i++)c[i]+=c[i-1];
for(int i=len;i>=1;i--)sa[c[x[y[i]]]--]=y[i],y[i]=0;
for(int i=1;i<=len;i++)swap(x[i],y[i]);
x[sa[1]]=1;num=1;
for(int i=2;i<=len;i++)
x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
if(num==len)break;
m=num;
}
}
void get_height(){
int k=0;
for(int i=1;i<=len;i++)rk[sa[i]]=i;
for(int i=1;i<=len;i++){
if(rk[i]==1)continue;
if(k)k--;
int j=sa[rk[i]-1];
while(i+k<=len&&j+k<=len&&s[i+k]==s[j+k])k++;
height[rk[i]]=k;
}
}
void pre_work(){
for(int i=1;i<=len;i++)mn[i][0]=height[i];
int Len=log2(len);
for(int j=1;j<=Len;j++)
for(int i=1;i+(1<<j)-1<=len;i++)
mn[i][j]=min(mn[i][j-1],mn[i+(1<<j-1)][j-1]);
}
int get_lcp(int l,int r){
if(l>r)swap(l,r);
l++;
if(l>r)return 1e9;
int Len=log2(r-l+1);
return min(mn[l][Len],mn[r-(1<<Len)+1][Len]);
}
void work(int x,int y,int k){
while(top[x]!=top[y]){
if(dep[top[x]]>dep[top[y]]){
Num[k]++;
L[Num[k]][k]=id[dfn[x]][1];
R[Num[k]][k]=id[dfn[top[x]]][1];
x=fa[top[x]];
}
else{
NUM[k]--;
L[NUM[k]][k]=id[dfn[top[y]]][0];
R[NUM[k]][k]=id[dfn[y]][0];
y=fa[top[y]];
}
}
if(dep[x]>dep[y]){
Num[k]++;
L[Num[k]][k]=id[dfn[x]][1];
R[Num[k]][k]=id[dfn[y]][1];
}
else{
NUM[k]--;
L[NUM[k]][k]=id[dfn[x]][0];
R[NUM[k]][k]=id[dfn[y]][0];
}
}
void calc(){
int now0=1,now1=1;
int tmp=0;
while(now0<=n&&now1<=n){
if(now0>Num[0]&&now0<NUM[0])now0=NUM[0];
if(now1>Num[1]&&now1<NUM[1])now1=NUM[1];
if(now0>n||now1>n)break;
int LL=min(min(R[now0][0]-L[now0][0]+1,R[now1][1]-L[now1][1]+1),get_lcp(rk[L[now0][0]],rk[L[now1][1]]));
L[now0][0]=L[now0][0]+LL;
L[now1][1]=L[now1][1]+LL;
tmp+=LL;
if(L[now0][0]<=R[now0][0]&&L[now1][1]<=R[now1][1])break;
if(L[now0][0]>R[now0][0])now0++;
if(L[now1][1]>R[now1][1])now1++;
}
printf("%d\n",tmp);
}
int read(){
int sum=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){sum=sum*10+ch-'0';ch=getchar();}
return sum*f;
}
int main(){
n=read();
scanf("%s",S+1);
for(int i=1;i<n;i++){
int u=read(),v=read();
add(u,v);add(v,u);
}
dfs1(1,0);cn=1;Top[1]=1;dfs2(1,1);
for(int i=1;i<=cn;i++)build(Top[i]);
m=122;
get_sa();get_height();pre_work();
m=read();
int a,b,c,d;
while(m--){
a=read();b=read();
c=read();d=read();
Num[0]=Num[1]=0;
NUM[0]=NUM[1]=n+1;
work(a,b,0);work(c,d,1);
calc();
}
return 0;
}
CF504E Misha and LCP on Tree(树链剖分+后缀树组)的更多相关文章
- 树链剖分 + 后缀数组 - E. Misha and LCP on Tree
E. Misha and LCP on Tree Problem's Link Mean: 给出一棵树,每个结点上有一个字母.每个询问给出两个路径,问这两个路径的串的最长公共前缀. analyse: ...
- Aizu 2450 Do use segment tree 树链剖分+线段树
Do use segment tree Time Limit: 1 Sec Memory Limit: 256 MB 题目连接 http://www.bnuoj.com/v3/problem_show ...
- 【POJ3237】Tree(树链剖分+线段树)
Description You are given a tree with N nodes. The tree’s nodes are numbered 1 through N and its edg ...
- POJ3237 Tree 树链剖分 线段树
欢迎访问~原文出处——博客园-zhouzhendong 去博客园看该题解 题目传送门 - POJ3237 题意概括 Description 给你由N个结点组成的树.树的节点被编号为1到N,边被编号为1 ...
- 【CF725G】Messages on a Tree 树链剖分+线段树
[CF725G]Messages on a Tree 题意:给你一棵n+1个节点的树,0号节点是树根,在编号为1到n的节点上各有一只跳蚤,0号节点是跳蚤国王.现在一些跳蚤要给跳蚤国王发信息.具体的信息 ...
- Spoj Query on a tree SPOJ - QTREE(树链剖分+线段树)
You are given a tree (an acyclic undirected connected graph) with N nodes, and edges numbered 1, 2, ...
- Water Tree CodeForces 343D 树链剖分+线段树
Water Tree CodeForces 343D 树链剖分+线段树 题意 给定一棵n个n-1条边的树,起初所有节点权值为0. 然后m个操作, 1 x:把x为根的子树的点的权值修改为1: 2 x:把 ...
- 【BZOJ-2325】道馆之战 树链剖分 + 线段树
2325: [ZJOI2011]道馆之战 Time Limit: 40 Sec Memory Limit: 256 MBSubmit: 1153 Solved: 421[Submit][Statu ...
- POJ3237 (树链剖分+线段树)
Problem Tree (POJ3237) 题目大意 给定一颗树,有边权. 要求支持三种操作: 操作一:更改某条边的权值. 操作二:将某条路径上的边权取反. 操作三:询问某条路径上的最大权值. 解题 ...
随机推荐
- Unity的SendMessage方法
用法(该对象所有脚本都能收到): gameObject.SendMessage("要执行的方法名"); 通知的另一种实现: gameObject.GetComponent<脚 ...
- Unity 导入的模型检测不到碰撞
解决方案 添加Mesh Collider和rigidbody,并且Collider里边2个勾要勾上. 这是我碰到的问题.如果没解决你的问题,别打我,逃~
- day05-1 执行Python程序的两种方式
目录 执行Python程序的两种方式 第一种:交互式 第二种:命令行式 三个步骤 两种方式的区别 执行Python程序的两种方式 第一种:交互式 在cmd中运行 优点:直接给出结果,执行效率高,及时报 ...
- Java实现18位身份证校验代码
import java.util.Scanner; /** * 18位身份证校验 * @author [J.H] * */ public class Test { // 身份证校验 public st ...
- 路飞学城Python-Day37(practise)
#1.自行创建测试数据 create database homework; use homework; # 年级表->老师表->课程表->班级表->学生表->成绩表-&g ...
- IOS - plist使用
//1 可读取,不可写入工程下的plist文件: // NSString *plistPath = [[NSBundle mainBundle] pathForResource:@"O ...
- 一、Git起步
1.关于版本控制 版本控制系统:版本控制是一种记录一个或若干文件内容变化,以便将来查阅特定版本修订情况的系统.但实际上,你可以对任何类型的文件进行版本控制. 1.1 本地版本控制系统 许多人习惯用复制 ...
- Linux red hat 核心版下安装Nginx
不要安装核心版的Linux,不要安装核心版的Linux,不要安装核心版的Linux重要的事情要说3遍.心血来潮突然想在Linux下安装Nginx,但是在安装的国程中发现了很多问题.nginx 基本安装 ...
- W10子系统UBantu命令安装Redis及其启动
W10子系统UBantu命令安装Redis及其启动 打开W10子系统UBantu 安装Redis $sudo apt-get install redis-server 启动Redis redis-se ...
- 20个C#热点问题