[HAOI2016] 找相同字符 - 后缀数组,单调栈
[HAOI2016] 找相同字符
Description
给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数。两个方案不同当且仅当这两个子串中有一个位置不同。 \(n,m \le 200000\)
Solution
将两个字符串按序连接,中间用一个其它字符隔开,新串记为 \(S\) ,那么 \(S[1 , n]\) 为第一个字符串的对应部分, \(S[n+2 ,n+m+1]\) 为第二个字符串对应的部分。预处理出 \(S\) 的后缀数组和高度数组。
我们将这些后缀中起始位置在 \([1,n]\) 内的称为黑后缀, \([n+2,n+m+1]\) 内的成为白后缀。那么我们可以考虑对每个白后缀它与所有黑后缀匹配的答案,这个贡献就是它与所有黑后缀的 \(LCP\) 长度的和。
假设所有后缀的顺序是按后缀排序的,不难想到分为左边的黑后缀和右边的黑后缀两部分,分开处理。
对于所黑后缀在白后缀左边的答案,我们可以按照顺序扫描所有的白后缀,同时维护到当前位置为止,当前位置串与前面任意一个位置串的 \(LCP\) 长度,同时记录它们的和。不难发现这可以用一个单调栈来处理。具体地,我们维护一个单调递增的栈,在栈中需要记录每个元素的下标,配合一个描述黑白后缀的前缀和数组,这样可以快速更新答案。
每当我们扫描到一个白后缀,就把单调栈中的和加进答案。
同理对所有黑后缀在白后缀右边的答案也去这样处理一遍。得到的就是最终的结果。
实现层面上这个题还是挺水了(虽然还是挠了半天毛)
Code
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 500005;
#define reset(x) memset(x,0,sizeof x)
namespace SA
{
int n,m=256,sa[N],y[N],u[N],v[N],o[N],r[N],h[N],T;
char str[N];
void solve()
{
n=strlen(str+1);
for(int i=1; i<=n; i++)
u[str[i]]++;
for(int i=1; i<=m; i++)
u[i]+=u[i-1];
for(int i=n; i>=1; i--)
sa[u[str[i]]--]=i;
r[sa[1]]=1;
for(int i=2; i<=n; i++)
r[sa[i]]=r[sa[i-1]]+(str[sa[i]]!=str[sa[i-1]]);
for(int l=1; r[sa[n]]<n; l<<=1)
{
memset(u,0,sizeof u);
memset(v,0,sizeof v);
memcpy(o,r,sizeof r);
for(int i=1; i<=n; i++)
u[r[i]]++, v[r[i+l]]++;
for(int i=1; i<=n; i++)
u[i]+=u[i-1], v[i]+=v[i-1];
for(int i=n; i>=1; i--)
y[v[r[i+l]]--]=i;
for(int i=n; i>=1; i--)
sa[u[r[y[i]]]--]=y[i];
r[sa[1]]=1;
for(int i=2; i<=n; i++)
r[sa[i]]=r[sa[i-1]]+((o[sa[i]]!=o[sa[i-1]])||(o[sa[i]+l]!=o[sa[i-1]+l]));
}
{
int i,j,k=0;
for(int i=1; i<=n; h[r[i++]]=k)
for(k?k--:0,j=sa[r[i]-1]; str[i+k]==str[j+k]; k++);
}
}
}
namespace MS
{
int a[N],b[N],c[N],p,ans;
void init()
{
reset(a);
reset(b);
reset(c);
p=0;
ans=0;
}
void push(int x,int y)
{
while(x<=a[p]&&p)
{
ans-=(c[b[p]]-c[b[p-1]])*a[p];
--p;
}
a[++p]=x;
b[p]=y;
ans+=(c[b[p]]-c[b[p-1]])*x;
}
}
char a[N],b[N];
int n,m,t1,t2,t3,t4,ans;
signed main()
{
cin>>a+1>>b+1;
n=strlen(a+1);
m=strlen(b+1);
reset(SA::str);
for(int i=1; i<=n; i++)
SA::str[i]=a[i];
SA::str[n+1]='$';
for(int i=1; i<=m; i++)
SA::str[n+i+1]=b[i];
SA::solve();
for(int i=1; i<=n+m+1; i++)
MS::c[i+1]=(SA::sa[i]<=n?1:0);
for(int i=1; i<=n+m+1; i++)
MS::c[i]+=MS::c[i-1];
for(int i=1; i<=n+m+1; i++)
{
MS::push(SA::h[i],i);
if(SA::sa[i]<=n+1)
continue;
ans+=MS::ans;
}
MS::init();
for(int i=1; i<=n+m+1; i++)
MS::c[i+1]=(SA::sa[i]>n+1?1:0);
for(int i=1; i<=n+m+1; i++)
MS::c[i]+=MS::c[i-1];
for(int i=1; i<=n+m+1; i++)
{
MS::push(SA::h[i],i);
if(SA::sa[i]>n)
continue;
ans+=MS::ans;
}
cout<<ans<<endl;
}
[HAOI2016] 找相同字符 - 后缀数组,单调栈的更多相关文章
- 【BZOJ4566】[Haoi2016]找相同字符 后缀数组+单调栈
[BZOJ4566][Haoi2016]找相同字符 Description 给定两个字符串,求出在两个字符串中各取出一个子串使得这两个子串相同的方案数.两个方案不同当且仅当这两 个子串中有一个位置不同 ...
- BZOJ.4566.[HAOI2016]找相同字符(后缀数组 单调栈)
题目链接 给定两个字符串,求它们有多少个相同子串.相同串的位置不同算多个. POJ3145简化版. 后缀自动机做法见这儿,又快又好写(一下就看出差距了..) //13712kb 4076ms #inc ...
- 【BZOJ-3238】差异 后缀数组 + 单调栈
3238: [Ahoi2013]差异 Time Limit: 20 Sec Memory Limit: 512 MBSubmit: 1561 Solved: 734[Submit][Status] ...
- BZOJ_3238_[Ahoi2013]差异_后缀数组+单调栈
BZOJ_3238_[Ahoi2013]差异_后缀数组+单调栈 Description Input 一行,一个字符串S Output 一行,一个整数,表示所求值 Sample Input cacao ...
- BZOJ_3879_SvT_后缀数组+单调栈
BZOJ_3879_SvT_后缀数组+单调栈 Description (我并不想告诉你题目名字是什么鬼) 有一个长度为n的仅包含小写字母的字符串S,下标范围为[1,n]. 现在有若干组询问,对于每一个 ...
- BZOJ.4199.[NOI2015]品酒大会(后缀数组 单调栈)
BZOJ 洛谷 后缀自动机做法. 洛谷上SAM比SA慢...BZOJ SAM却能快近一倍... 显然只需要考虑极长的相同子串的贡献,然后求后缀和/后缀\(\max\)就可以了. 对于相同子串,我们能想 ...
- 【BZOJ3879】SvT 后缀数组+单调栈
[BZOJ3879]SvT Description (我并不想告诉你题目名字是什么鬼) 有一个长度为n的仅包含小写字母的字符串S,下标范围为[1,n]. 现在有若干组询问,对于每一个询问,我们给出若干 ...
- BZOJ3238 [Ahoi2013]差异 【后缀数组 + 单调栈】
题目链接 BZOJ3238 题解 简单题 经典后缀数组 + 单调栈套路,求所有后缀\(lcp\) #include<iostream> #include<cstdio> #in ...
- BZOJ4199 [Noi2015]品酒大会 【后缀数组 + 单调栈 + ST表】
题目 一年一度的"幻影阁夏日品酒大会"隆重开幕了.大会包含品尝和趣味挑战两个环节,分别向优胜者颁发"首席品 酒家"和"首席猎手"两个奖项,吸 ...
随机推荐
- 有多少人在面试时,被Java 如何线程间通讯,问哭了?
正常情况下,每个子线程完成各自的任务就可以结束了.不过有的时候,我们希望多个线程协同工作来完成某个任务,这时就涉及到了线程间通信了. 本文涉及到的知识点: thread.join(), object. ...
- 战“疫”背后的AI身影丨曼孚科技
近期新型冠状病毒肺炎的疫情,牵动着全国上下人民的心. 截止2月11日上午10点,全国确诊人数已达42708人,疑似病例21675人. 突发的疫情让部分地区的快速诊疗能力出现了结构性的缺失,为了打赢这场 ...
- 当课堂因监控技术变“囚笼”,存在争议的AI使用场景该被抵制吗?
当马云和马斯克高谈阔论AI是否会影响人类社会时,尚无"感情"的AI已在校园中"作恶".近日,一张AI监控课堂的GIF在网上迅速刷屏.这张GIF中记录了课堂中所有 ...
- 写Blog的准备工作Window Live Writer
以下是博客园的帮助博客: Windows Live Writer配置步骤 感觉里面的几篇推荐文档都可以看一下,至于代码高亮插件,个人选择的还是SyntaxHighlighter. 一. 关于写blog ...
- 网络共享服务(一)之FTP
网络共享服务:ftp,nfs,samba比较 从跨平台角度说, samba, ftp差不多, 而nfs不支持windows平台 从挂载角度说, samba, nfs可以把远程目录挂载到本地目录上, 对 ...
- CF840D[Destiny] [主席树板子]
模板题,提供两种思路. 1.考虑它求得是 出现个数\(>\lfloor \frac{r-l+1}{k}\rfloor\) 的最小值 那么我们可以根据如果大于这个数那么你把这个区间数列排好序然后分 ...
- 遇到的错误:Mysql 报错Duplicate entry '值' for key '字段名'的解决
### Error updating database. Cause: com.mysql.jdbc.exceptions.jdbc4.MySQLIntegrityConstraintViolatio ...
- linux-mysql-主从同步
什么是二进制日志binlog:记录着mysql数据库中的一些写入性操作,比如一些增删改,但不包括查询!二进制日志有哪些功能:数据复制和数据恢复的功能 查看网络状态:netstat -natp查看mas ...
- Help Jimmy POJ - 1661 dp
#include<iostream> #include<stdio.h> #include<algorithm> #include<cstring> u ...
- AOPS论坛上100+100个积分
100+10 rare and irresistible integrals I bring you many beautiful integrals that I have collected ov ...