icpc2018焦作H.Can You Solve the Harder Problem?

题意

求一个串的本质不同子串的权值和，一个串的权值为这个串所有元素的最大值

思路

如果不考虑本质不同子串，可以设第$i$个元素为左端点的所有子串权值和为$f_i$,那么假设$i$左边第一个比$i$位置大的元素位置为$j$，那么可得如下等式
$f_i=f_j+num[i]\ast (j-i)$
单调栈处理每个数左边第一个比他大的地方，再反向枚举$i$即可

现在考虑本质不同子串，很容易想到后缀数组，$height[Rank[i]]$就是后缀$i$与它前一名的后缀重复的前缀，也就是重复的子串，这一部分是不算贡献的，首先如果$height[Rank[i]]$的值为0，那么结果直接加上$f_i$，因为和前一名没有公共前缀，也就是和前一名没有重复子串，否则如下图

求蓝色部分的值，$j$为在$i$到$i+height[i]-1$这个区间的最大值的位置，$k$为第一个比$j$大的地方,那么蓝色部分的贡献就是：
$f_k+num[j]\ast (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!!!