[BZOJ4504][主席树]K个串

只区间出现一次的数就是上一次出现在位置在区间外的数，这个可以用主席树实现，然后套用超级钢琴的思想就可以了。

#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;
}