[洛谷P3181][HAOI2016]找相同字符(后缀数组)

把两个字符串连起来,后缀数组求出height数组后用单调栈维护。记录排名i的后缀的height可以向前延伸到最远时,其间l[i]个后缀属于字符串1,r[i]个后缀属于字符串2。后缀i每弹出一个栈内元素,贡献为当前的l[i]*r[i]再乘弹出元素与前一个栈内元素的差或与height[i]的差的最小值。注意后缀i的l和r数组初始值应由后缀i-1属于哪个字符串决定。

#include<cstdio> 
#include<cstring>
#include<algorithm>
using namespace std;
const int N=400010;
int n1,n,m,rnk[N],tp[N],sa[N],hgt[N],tak[N],top,q[N],pos[N],l[N],r[N];
char str1[N];
long long ans;
inline void sort1(){
	for(int i=1;i<=m;i++)tak[i]=0;
	for(int i=1;i<=n;i++)tak[rnk[i]]++;
	for(int i=2;i<=m;i++)tak[i]+=tak[i-1];
	for(int i=n;i>=1;i--)sa[tak[rnk[tp[i]]]--]=tp[i];
}
int main(){
	scanf("%s",str1+1);
	n=n1=strlen(str1+1);
	for(int i=1;i<=n1;i++){rnk[i]=str1[i]-'a'+1;tp[i]=i;}
	str1[n+1]='&';rnk[n+1]=27;tp[n+1]=n+1;
	scanf("%s",str1+n1+2);n=strlen(str1+1);
	for(int i=n1+2;i<=n;i++){rnk[i]=str1[i]-'a'+1;tp[i]=i;}
	m=27;sort1();
	for(int p=0,w=1;p<n;w<<=1,m=p){
		p=0;
		for(int i=1;i<=w;i++)tp[++p]=n-w+i;
		for(int i=1;i<=n;i++)if(sa[i]>w)tp[++p]=sa[i]-w;
		sort1();swap(tp,rnk);
		rnk[sa[1]]=p=1;
		for(int i=2;i<=n;i++){
			if(tp[sa[i]]!=tp[sa[i-1]]||tp[sa[i]+w]!=tp[sa[i-1]+w])p++;
			rnk[sa[i]]=p;
		}
	}
	for(int x,j=0,i=1;i<=n;i++){
		if(j)j--;
		x=sa[rnk[i]-1];
		while(str1[i+j]==str1[x+j])j++;
		hgt[rnk[i]]=j;
	}
	ans=0;
	memset(l,0,sizeof l);memset(r,0,sizeof r);
	top=1;q[0]=q[top]=0;pos[1]=1;
	for(int i=2;i<=n;i++){
		if(sa[i-1]<=n1)l[i]=1;
		else r[i]=1;
		while(top>0&&hgt[i]<=q[top]){
			l[i]+=l[pos[top]];r[i]+=r[pos[top]];
			ans+=1ll*l[i]*r[i]*(q[top]-max(q[top-1],hgt[i]));
			top--;
		}
		q[++top]=hgt[i];pos[top]=i;
	}
	printf("%lld",ans);
	return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
这道题目还可以使用树状数组或线段树来实现,时间复杂度也为 $\mathcal{O}(n\log n)$。这里给出使用树状数组的实现代码。 解题思路: 1. 读入数据; 2. 将原数列离散化,得到一个新的数列 b; 3. 从右往左依次将 b 数列中的元素插入到树状数组中,并计算逆序对数; 4. 输出逆序对数。 代码实现: ```c++ #include <cstdio> #include <cstdlib> #include <algorithm> const int MAXN = 500005; struct Node { int val, id; bool operator<(const Node& other) const { return val < other.val; } } nodes[MAXN]; int n, a[MAXN], b[MAXN], c[MAXN]; long long ans; inline int lowbit(int x) { return x & (-x); } void update(int x, int val) { for (int i = x; i <= n; i += lowbit(i)) { c[i] += val; } } int query(int x) { int res = 0; for (int i = x; i > 0; i -= lowbit(i)) { res += c[i]; } return res; } int main() { scanf("%d", &n); for (int i = 1; i <= n; ++i) { scanf("%d", &a[i]); nodes[i] = {a[i], i}; } std::sort(nodes + 1, nodes + n + 1); int cnt = 0; for (int i = 1; i <= n; ++i) { if (i == 1 || nodes[i].val != nodes[i - 1].val) { ++cnt; } b[nodes[i].id] = cnt; } for (int i = n; i >= 1; --i) { ans += query(b[i] - 1); update(b[i], 1); } printf("%lld\n", ans); return 0; } ``` 注意事项: - 在对原数列进行离散化时,需要记录每个元素在原数列中的位置,便于后面计算逆序对数; - 设树状数组的大小为 $n$,则树状数组中的下标从 $1$ 到 $n$,而不是从 $0$ 到 $n-1$; - 在计算逆序对数时,需要查询离散化后的数列中比当前元素小的元素个数,即查询 $b_i-1$ 位置上的值; - 在插入元素时,需要将离散化后的数列的元素从右往左依次插入树状数组中,而不是从左往右。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值