BZOJ 5011 [JXOI2016]颜色

---

题目描述 传送门

---

按照老师的思路：倍增(st表)+递推 完成的此题。
但不知道是不是我没写好还是我思路不好，这份代码在BZOJ明显比别人的慢。
希望dalao能分享更好的解法。
代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=3e5+5;
int n;
int a[maxn],begin[maxn],end[maxn],d[maxn][20],f[maxn][20];
long long dp[maxn];
inline int rmq(int l,int r,int o){
    int k=0;
    while(1<<(k+1)<=r-l+1) k++;
    return o?max(f[l][k],f[r-(1<<k)+1][k]):min(d[l][k],d[r-(1<<k)+1][k]);
}
int main(){
    int t;
    cin>>t;
    while(t--){
        scanf("%d",&n);
        memset(begin,0,sizeof(begin));
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            if(!begin[a[i]]) begin[a[i]]=end[a[i]]=i;
            else end[a[i]]=i;
        }
        for(int i=1;i<=n;i++) d[i][0]=begin[a[i]],f[i][0]=end[a[i]];
        for(int j=1;(1<<j)<=n;j++)
            for(int i=1;i+(1<<j)-1<=n;i++){
                d[i][j]=min(d[i][j-1],d[i+(1<<(j-1))][j-1]);
                f[i][j]=max(f[i][j-1],f[i+(1<<(j-1))][j-1]);    
            }
        dp[n+1]=0;  
        long long ans=0;
        for(int i=n;i>0;i--){
            if(i!=begin[a[i]]) dp[i]=0;
            else{
                int L=begin[a[i]],R=end[a[i]];
                bool ok=1;
                for(;;){     //找到以i为起点的最短可选区间[i,R]
                    if(rmq(L,R,0)<i) {ok=0;break;}
                    int k=rmq(L,R,1);
                    if(k>R){
                        L=R+1;
                        R=k;
                    }
                    else break;
                }
                if(ok){
                    if(i==1&&(R==n||dp[R+1]>0)) ans--;    //减去全选的
                    dp[i]=dp[R+1]+1;   //从后往前递推
                }
            }
        }
        for(int i=1;i<=n;i++) ans+=dp[i];
        printf("%lld\n",ans+1);
    }
    return 0;
}