只区间出现一次的数就是上一次出现在位置在区间外的数,这个可以用主席树实现,然后套用超级钢琴的思想就可以了。
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <map>
#include <queue>
using namespace std;
const int N=100010;
typedef long long ll;
typedef pair<ll,int> parli;
int n,k,cnt,a[N],lst[N],rt[N];
int ls[N*50],rs[N*50];
ll add[N*50];
parli mx[N*50];
struct stp{
ll val; int l,r,p,x;
friend bool operator <(stp a,stp b){
return a.val<b.val;
}
};
map<ll,int> A;
priority_queue<stp> Q;
void Add(int lg,int &g,int l,int r,int x,int L,int R){
g=++cnt; mx[g]=mx[lg]; ls[g]=ls[lg]; rs[g]=rs[lg]; add[g]=add[lg];
if(l==L&&r==R){
mx[g].first+=x; add[g]+=x;
return ;
}
int mid=L+R>>1;
if(r<=mid) Add(ls[lg],ls[g],l,r,x,L,mid);
else if(l>mid) Add(rs[lg],rs[g],l,r,x,mid+1,R);
else Add(ls[lg],ls[g],l,mid,x,L,mid),Add(rs[lg],rs[g],mid+1,r,x,mid+1,R);
mx[g]=max(mx[ls[g]],mx[rs[g]]); mx[g].first+=add[g];
}
parli Query(int g,int l,int r,int L,int R){
if(!g) return parli(-1e9,0);
if(l==L&&r==R) return mx[g];
int mid=L+R>>1; parli ret;
if(r<=mid) ret=Query(ls[g],l,r,L,mid);
else if(l>mid) ret=Query(rs[g],l,r,mid+1,R);
else ret=max(Query(ls[g],l,mid,L,mid),Query(rs[g],mid+1,r,mid+1,R));
ret.first+=add[g]; return ret;
}
void Build(int &g,int l,int r){
g=++cnt;
if(l==r){ mx[g].first=0; mx[g].second=r; return; }
int mid=l+r>>1;
Build(ls[g],l,mid); Build(rs[g],mid+1,r);
mx[g]=max(mx[ls[g]],mx[rs[g]]);
}
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
if(A.count(a[i])) lst[i]=A[a[i]];
A[a[i]]=i;
}
Build(rt[0],1,n);
for(int i=1;i<=n;i++){
rt[i]=rt[i-1];
Add(rt[i-1],rt[i],lst[i]+1,i,a[i],1,n);
parli q=Query(rt[i],1,i,1,n);
Q.push((stp){q.first,1,i,q.second,i});
}
for(int i=1;i<k;i++){
stp cur=Q.top(); Q.pop();
if(cur.l==cur.r) continue;
parli q;
if(cur.l<cur.p){
q=Query(rt[cur.x],cur.l,cur.p-1,1,n);
Q.push((stp){q.first,cur.l,cur.p-1,q.second,cur.x});
}
if(cur.p<cur.r){
q=Query(rt[cur.x],cur.p+1,cur.r,1,n);
Q.push((stp){q.first,cur.p+1,cur.r,q.second,cur.x});
}
}
printf("%lld\n",Q.top().val);
return 0;
}