【BZOJ4504&&Hihocoder1046】K个串（主席树，堆）

题意：一个长度为n的数字序列，选出其中的一个连续子串，然后统计其子串中所有数字之和（注意这里重复出现的数字只被统计一次）询问第k大的和是多少

1 <= n <= 100000, 1 <= k <= 200000, 0 <= a[i]<= 10^9

思路：类似于超级钢琴的思路

对于每一个右端点建立一棵主席树维护左端点的最大值

对于a[i]它对[last[a[i]]+1,i]有a[i]的贡献

再维护一个堆维护全局最大值

删除的时候就找到最大值的位置赋值成最小值

C++ STL 不会用 好烦

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<ctime>
 #include<cmath>
 #include<algorithm>
 #include<iomanip>
 #include<vector>
 #include<map>
 #include<set>
 #include<bitset>
 #include<queue>
 #include<stack>
 using namespace std;
 typedef long long ll;
 typedef unsigned int uint;
 typedef unsigned long long ull;
 typedef pair<int,int> PII;
 typedef vector<int> VI;
 #define fi first
 #define se second
 #define MP make_pair
 #define N   21000000
 #define M   110000
 #define eps 1e-8
 #define pi  acos(-1)
 #define oo  1e18
 #define MOD 10007

 struct node
 {
     ll v;
     int id;
     node()
     {
     }
     node(ll V,int P)
     {
         v=V; id=P;
     }
     friend bool operator <(node x,node y)
     {
         return x.v<y.v;
     }
 };

 struct tree
 {
     int l,r;
     ll a,s;
 }t[N];

 priority_queue<node>q;
 map<int,int>last;
 int a[M],root[M],n,cnt;

 void pushup(int p)
 {
     t[p].s=;
     if(!t[p].r) t[p].s=t[t[p].l].s;
      else if(!t[p].l) t[p].s=t[t[p].r].s;
       else t[p].s=max(t[t[p].l].s,t[t[p].r].s);
     t[p].s+=t[p].a;
 }

 void update(int l,int r,int x,int y,int v,int &r1,int r2)
 {
     r1=++cnt;
     t[r1]=t[r2];
     if(x<=l&&r<=y)
     {
         t[r1].a+=v;
         pushup(r1);
         return;
     }
     int mid=(l+r)>>;
     if(x<=mid) update(l,mid,x,y,v,t[r1].l,t[r2].l);
     if(y>mid) update(mid+,r,x,y,v,t[r1].r,t[r2].r);
     pushup(r1);
 }

 void del(int l,int r,int &p)
 {
     int x=p;
     p=++cnt;
     t[cnt]=t[x];
     if(l==r)
     {
         t[p].a=-oo;
         pushup(p);
         return;
     }
     int mid=(l+r)>>;
     if(!t[p].r) del(l,mid,t[p].l);
      else if(!t[p].l) del(mid+,r,t[p].r);
       else
       {
               if(t[t[p].l].s>=t[t[p].r].s) del(l,mid,t[p].l);
                else del(mid+,r,t[p].r);
       }
     pushup(p);
 }

 int main()
 {
     //freopen("bzoj4504.in","r",stdin);
     //freopen("bzoj4504.out","w",stdout);
     int k;
     scanf("%d%d",&n,&k);
     cnt=;
     for(int i=;i<=n;i++)
     {
         scanf("%d",&a[i]);
         update(,n,i,i,,root[i],root[i-]);
         update(,n,last[a[i]]+,i,a[i],root[i],root[i]);
         last[a[i]]=i;
         q.push(node(t[root[i]].s,i));
     }
     for(int i=;i<k;i++)
     {
         int x=q.top().id;
         q.pop();
         del(,n,root[x]);
         q.push(node(t[root[x]].s,x));
     }
     printf("%lld\n",q.top().v);
     return ;
 }