题意
求一个串的本质不同子串的权值和,一个串的权值为这个串所有元素的最大值
思路
如果不考虑本质不同子串,可以设第
i
i
i个元素为左端点的所有子串权值和为
f
i
f_i
fi,那么假设
i
i
i左边第一个比
i
i
i位置大的元素位置为
j
j
j,那么可得如下等式
f
i
=
f
j
+
n
u
m
[
i
]
∗
(
j
−
i
)
f_i=f_j+num[i]*(j-i)
fi=fj+num[i]∗(j−i)
单调栈处理每个数左边第一个比他大的地方,再反向枚举
i
i
i即可
现在考虑本质不同子串,很容易想到后缀数组,
h
e
i
g
h
t
[
R
a
n
k
[
i
]
]
height[Rank[i]]
height[Rank[i]]就是后缀
i
i
i与它前一名的后缀重复的前缀,也就是重复的子串,这一部分是不算贡献的,首先如果
h
e
i
g
h
t
[
R
a
n
k
[
i
]
]
height[Rank[i]]
height[Rank[i]]的值为0,那么结果直接加上
f
i
f_i
fi,因为和前一名没有公共前缀,也就是和前一名没有重复子串,否则如下图
求蓝色部分的值,
j
j
j为在
i
i
i到
i
+
h
e
i
g
h
t
[
i
]
−
1
i+height[i]-1
i+height[i]−1这个区间的最大值的位置,
k
k
k为第一个比
j
j
j大的地方,那么蓝色部分的贡献就是:
f
k
+
n
u
m
[
j
]
∗
(
k
−
i
−
h
e
i
g
h
t
[
R
a
n
k
[
i
]
]
)
f_k+num[j]*(k-i-height[Rank[i]])
fk+num[j]∗(k−i−height[Rank[i]]),区间最大值的位置可以用线段树来维护
代码
#include <bits/stdc++.h>
#pragma gcc optimize(2)
using namespace std;
#define ll long long
const int maxn = 2e5 + 10;
#define endl '\n'
int wa[maxn], wb[maxn], Ws[maxn], R[maxn];
int Rank[maxn], height[maxn], sa[maxn];
ll sum[maxn];
bool cmp(int *s, int a, int b, int l)
{
return s[a] == s[b] && s[a + l] == s[b + l];
}
void get_sa(int n,int m){
int i,j,p,*x=wa,*y=wb,*t;
for(i=0;i<m;i++) Ws[i]=0;
for(i=0;i<n;i++) Ws[x[i]=R[i]]++;
for(i=1;i<m;i++) Ws[i]+=Ws[i-1];
for(i=n-1;i>=0;i--) sa[--Ws[x[i]]]=i;
for(j=1,p=1;p<n;j*=2,m=p){
for(p=0,i=n-j;i<n;i++) y[p++]=i;
for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
for(i=0;i<m;i++) Ws[i]=0;
for(i=0;i<n;i++) Ws[x[y[i]]]++;
for(i=1;i<m;i++) Ws[i]+=Ws[i-1];
for(i=n-1;i>=0;i--) sa[--Ws[x[y[i]]]]=y[i];
for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++){
x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
}
}
return ;
}
void get_height(int n){
int i,j,k=0;
for(i=1;i<=n;i++) Rank[sa[i]]=i;
for(i=0;i<n;i++){
if(k) k--;
j=sa[Rank[i]-1];
while(R[i+k]==R[j+k]) k++;
height[Rank[i]]=k;
}
return ;
}
int x[maxn];
struct Sgt{
int l;
int r;
int mp;
};
Sgt sgt[maxn<<2];
void push_up(int p){
if(R[sgt[p<<1].mp-1]>=R[sgt[p<<1|1].mp-1]) sgt[p].mp=sgt[p<<1].mp;
else sgt[p].mp=sgt[p<<1|1].mp;
return ;
}
void build(int p,int l,int r){
sgt[p].l=l;
sgt[p].r=r;
if(l==r){
sgt[p].mp=l;
return ;
}
int mid=l+r>>1;
build(p<<1,l,mid);
build(p<<1|1,mid+1,r);
push_up(p);
return ;
}
int query(int p,int l,int r){
if(l<=sgt[p].l&&r>=sgt[p].r) return sgt[p].mp-1;
int mid=sgt[p].l+sgt[p].r>>1;
if(r<=mid) return query(p<<1,l,r);
if(l>mid) return query(p<<1|1,l,r);
int p1=query(p<<1,l,r);
int p2=query(p<<1|1,l,r);
if(R[p1]>=R[p2]) return p1;
else return p2;
}
int last[maxn];
stack<pair<int,int>>st;
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int t;
cin >> t;
while (t--)
{
int n;
cin >> n;
for (int i = 0; i < n; i++)
cin >> x[i], R[i] = x[i];
sort(x ,x + n);
int sz=unique(x,x+n)-x;
for(int i=0;i<n;i++){
R[i]=lower_bound(x,x+sz,R[i])-x+1;
}
R[n]=0;
get_sa(n+1,n+1);
get_height(n);
st.push(make_pair(R[0],0));
build(1,1,n);
for(int i=1;i<n;i++){
pair<int,int>nw=make_pair(R[i],i);
while(!st.empty()&&st.top().first<R[i]){
last[st.top().second]=i;
st.pop();
}
st.push(nw);
}
while(!st.empty()){
last[st.top().second]=n;
st.pop();
}
sum[n]=0;
for(int i=n-1;i>=0;i--){
sum[i]=sum[last[i]]+1ll*x[R[i]-1]*(last[i]-i);
}
ll ans=0;
for(int i=0;i<n;i++){
if(height[Rank[i]]==0) ans+=sum[i];
else{
int pos=query(1,i+1,i+1+height[Rank[i]]-1);
ans+=sum[last[pos]]+1ll*x[R[pos]-1]*(last[pos]-i-height[Rank[i]]);
}
}
cout<<ans<<endl;
}
return 0;
}
一定记得开long long!!!