【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=0;
    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)>>1;
    if(x<=mid) update(l,mid,x,y,v,t[r1].l,t[r2].l);
    if(y>mid) update(mid+1,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)>>1;
    if(!t[p].r) del(l,mid,t[p].l);
     else if(!t[p].l) del(mid+1,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+1,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=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        update(1,n,i,i,0,root[i],root[i-1]);
        update(1,n,last[a[i]]+1,i,a[i],root[i],root[i]);
        last[a[i]]=i;
        q.push(node(t[root[i]].s,i));
    }
    for(int i=1;i<k;i++)
    {
        int x=q.top().id;
        q.pop();
        del(1,n,root[x]);
        q.push(node(t[root[x]].s,x));
    }
    printf("%lld
",q.top().v);
    return 0;
}