POJ - 3415 Common Substrings（长度不小于K的公共子串个数）

Common Substrings

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
typedef long long ll;
// sa[i]: 排名是i位的是第几个后缀
// rk[i]: 第i个后缀的排名是多少
// height[i]: sa[i]与sa[i-1]
const int N=200010;
char s[N],s1[N],s2[N];
int rk[N],sa[N],cnt[N],height[N];
int x[N],y[N];
int n,m,K;
int n1,n2;
void rsort()// x[i] 第一关键字 y[i] 第二关键字 基数排序
{for(int i=1;i<=m;i++) cnt[i]=0;for(int i=1;i<=n;i++) cnt[x[i]]++;for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];for(int i=n;i;i--) sa[cnt[x[y[i]]]--]=y[i];
}
void SA()
{n=strlen(s+1);m=300;for(int i=1;i<=n;i++) x[i]=s[i],y[i]=i;rsort();for(int k=1;k<=n;k<<=1){int p=0;for(int i=n-k+1;i<=n;i++) y[++p]=i;// 第二关键字为空字符排在最前面for(int i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;rsort();swap(x,y);x[sa[1]]=1,p=1;for(int i=2;i<=n;i++)x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p);if(p==n) break;m=p;}for(int i=1;i<=n;i++) rk[sa[i]]=i;// 求heightfor(int i=1,j=0;i<=n;i++){if(j) --j;while(s[i+j]==s[sa[rk[i]-1]+j]) j++;height[rk[i]]=j;}
}
int init()
{n1=strlen(s1+1);n2=strlen(s2+1);for(int i=1;i<=n1;i++) s[i]=s1[i];s[n1+1]='*';for(int i=1;i<=n2;i++) s[i+n1+1]=s2[i];s[n1+n2+1+1]='\0';return (n1+n2+1);
}
int st[N][2];
ll solve()
{ll ans=0,tot=0;//当前栈里的后缀与将要入栈的后缀的公共字串的个数// A前 B后int tt=0;for(int i=1;i<=n;i++){if(height[i]<K){tt=0,tot=0;continue;}int cnt=0;// 与以后入栈的序列lcp是height[i]的数目if(sa[i-1]<=n1) // 属于第一个字符串{cnt++;tot+=height[i]-K+1;}while(tt&&height[i]<=st[tt][0]){tot-=1ll*(st[tt][0]-height[i])*st[tt][1];// 由于更小的height[i]导致原先栈中的lcp多算st[tt][0]-height[i]cnt+=st[tt][1];tt--;}st[++tt][0]=height[i];st[tt][1]=cnt;if(sa[i]>n1+1) ans+=tot;// 属于第二个字符串}// B前 A后tt=0;for(int i=1;i<=n;i++){if(height[i]<K){tt=0,tot=0;continue;}int cnt=0;if(sa[i-1]>n1+1) // 属于第一个字符串{cnt++;tot+=height[i]-K+1;}while(tt&&height[i]<=st[tt][0]){tot-=1ll*(st[tt][0]-height[i])*st[tt][1];cnt+=st[tt][1];tt--;}st[++tt][0]=height[i];st[tt][1]=cnt;if(sa[i]<=n1+1) ans+=tot;// 属于第二个字符串}return ans;
}
int main()
{while(scanf("%d",&K),K){scanf("%s%s",s1+1,s2+1);n=init();SA();printf("%lld\n",solve());}return 0;
}

POJ - 3415 Common Substrings（长度不小于K的公共子串个数）相关推荐

POJ 3415 Common Substrings
Description A substring of a string T is defined as: T(i, k)=TiTi+1...Ti+k-1, 1≤i≤i+k-1≤|T|. Given t ...
POJ - 3415 Common Substrings(后缀数组+单调栈)
题目链接:点击查看题目大意:给出两个字符串,再给出一个k,问两个字符串中长度大于等于k的公共子串有多少个(种类可重复) 题目分析:因为涉及到了子串问题,先用后缀数组跑出height数组来,接下来如果 ...
cdoj915-方老师的分身 II （长度不小于k的最短路）【spfa】
http://acm.uestc.edu.cn/#/problem/show/915 方老师的分身 II Time Limit: 10000/5000MS (Java/Others) Memo ...
poj 3261 后缀数组找反复出现k次的子串（子串能够重叠）
题目:http://poj.org/problem?id=3261 仍然是后缀数组的典型应用----后缀数组+lcp+二分做的蛮顺的,1A 可是大部分时间是在调试代码.由于模板的全局变量用混了,而自 ...
hdu5056(找相同字母不出现k次的子串个数)
题意: 给你一个字符串,然后问你这个字符串里面有多少个满足要求的子串,要求是每个子串相同字母出现的次数不能超过k. 思路: 这种题目做着比较有意思,而且不是很难(但自己还是嘚瑟,w ...
【HDU - 4417】Super Mario（查询区间小于K的数的个数，主席树）
题干: Mario is world-famous plumber. His "burly" figure and amazing jumping ability reminded ...
【POJ3415】 Common Substrings (SA+单调栈)
这道是求长度不小于 k 的公共子串的个数...很不幸,我又TLE了... 解法参考论文以及下面的链接 http://www.cnblogs.com/vongang/archive/2012/11/20 ...
POJ 3415 后缀数组+单调栈
题目大意: 给定A,B两种字符串,问他们当中的长度大于k的公共子串的个数有多少个这道题目本身理解不难,将两个字符串合并后求出它的后缀数组然后利用后缀数组求解答案这里一开始看题解说要用栈的思想,觉 ...
最长公共子串LCS (Longest Common Subsequence) 算法
三个方法都有所借鉴,但代码部分是自己试着写出来的,虽然最后的运行结果都是正确的,但此过程中难免会有考虑不周全的地方,如发现代码某些地方有误,欢迎指正.同时有新的想法,也可以提出! 采用顺序结构存储串, ...

POJ - 3415 Common Substrings（长度不小于K的公共子串个数）

Common Substrings

POJ - 3415 Common Substrings（长度不小于K的公共子串个数）相关推荐

最新文章

热门文章