[bzoj P4504] K个串

[bzoj P4504] K个串

【题目描述】

兔子们在玩k个串的游戏。首先，它们拿出了一个长度为n的数字序列，选出其中的一个连续子串，然后统计其子串中所有数字之和（注意这里重复出现的数字只被统计一次）。

兔子们想知道，在这个数字序列所有连续的子串中，按照以上方式统计其所有数字之和，第k大的和是多少。

【输入格式】

第一行，两个整数n和k，分别表示长度为n的数字序列和想要统计的第k大的和

接下里一行n个数a_i，表示这个数字序列

【输出格式】

一行一个整数，表示第k大的和

【样例输入】

7 5

3 -2 1 2 2 1 3 -2

【样例输出】

4

【数据范围】

对于20%的数据，1 <= n <= 2000

对于另外20%的数据，0 <= a_i <= 10^9

对于100%的数据，1 <= n <= 100000, 1 <= k <= 200000, 0 <= |a_i| <= 10^9

数据保证存在第k大的和

题外话：好久都没有用博客园啦，最近写博客都写在WordPress上面。没想到今天炸了。。只好先来cnblogs避一避了。

对于这一题我也是一脸懵逼式的弃疗。连想都没怎么想，丝毫没有办法。

结果——主席树+堆。也是看了题解才明白的。自己怎么也想不到这种方法。

主思路是这样的，维护一个五元组(v,l,r,x,p)表示一个状态，分别表示当前状态所对应区间的和(v)，左端点所在区间(l,r)，左端点的具体位置(x)，右端点的具体位置(p)。（这个思路骑士真的太难想到了，主要一个我觉得，习惯于考虑对称的东西，不会像这样，区间左右端点有别）

先不考虑如何如何构造或得到五元组。假设我们可以很快得到某一个特定的五元组。那如何得出第k大的和？

每一次从堆中取出最大值，也就是当前最大的和，然后做这样的事情：

构造五元组(maxsum_val(),l,p-1,maxsum_pos(),p)和(maxsum_val(),p+1,r,maxsum_pos(),p)，并将它们push入堆中。显然这是正确的。

那刚开始在堆里的是什么呢？当然是p=1~n时，左端点在某一点能使区间和最大的这个状态咯。

这个问题——涉及到区间修改，区间查询。肯定要用线段树咯。但是我们发现用线段树是无法处理对于不同的右端点p，查询某个区间最值的问题的。

所以我们对于每一种右端点p，建立一颗主席树，然后在对应的主席树上高即可。

code：

 #pragma GCC optimize(2)
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <map>
 #include <queue>
 #define LL long long
 #define mp make_pair
 #define pli pair <LL,int>

 using namespace std;

 ,M=;

 int n,m; map <int,int> pre;
 struct node {
     LL v; int x,l,r,p;
     node () {}
     node (LL _v,int _x,int _l,int _r,int _p) :
         v(_v),x(_x),l(_l),r(_r),p(_p) {}
     bool operator < (const node &o) const {
         return v<o.v;
     }
 };
 priority_queue <node> q;

 namespace TREE {
     int tot,rt[N],lc[M],rc[M]; pli w[M]; LL tag[M];
     #define mid (((l)+(r))>>1)
     #define ms(a,x) memset(a,x,sizeof a)
     inline ,ms(tag,);}
     inline void build (int &u,int l,int r) {
         w[u=++tot]=mp(,l);
         if (l==r) return;
         build(lc[u],l,mid),build(rc[u],mid+,r);
     }
     inline void insert (int &u,int v,LL z) {
         tag[u=++tot]=tag[v]+z;
         lc[u]=lc[v],rc[u]=rc[v],w[u]=w[v],w[u].first+=z;
     }
     inline void upload (int u) {
         w[u]=max(w[lc[u]],w[rc[u]]);
     }
     inline void download (int &u) {
         insert(lc[u],lc[u],tag[u]),insert(rc[u],rc[u],tag[u]);
         tag[u]=;
     }
     inline void modify (int &u,int v,int l,int r,int x,int y,LL z) {
         if (x<=l&&r<=y) {insert(u,v,z); return;}
         if (tag[u]) download(v);
         lc[u=++tot]=lc[v],rc[u]=rc[v],w[u]=w[v];
         if (x<=mid) modify(lc[u],lc[v],l,mid,x,y,z);
         ,r,x,y,z);
         upload(u);
     }
     inline pli query ()) {
         if (x<=l&&r<=y) return w[u];
         if (tag[u]) download(u);
         if (x<=mid) ret=query(lc[u],l,mid,x,y);
         ,r,x,y));
         return ret;
     }
 } using namespace TREE;

 inline void extend (int x,int l,int r) {
     if (l>r) return;
     pli nxt=query(x,,n,l,r);
     q.push(node(nxt.first,x,l,r,nxt.second));
 }

 int main () {
     scanf(],,n);
     ,x; i<=n; ++i) {
         scanf("%d",&x);
         modify(rt[i],rt[i-],,n,pre[x]+,i,(LL)x);
         extend(rt[i],,i),pre[x]=i;
     }
     node cur;
     for ( ; m; --m) {
         cur=q.top(),q.pop();
         extend(cur.x,cur.l,cur.p-);
         extend(cur.x,cur.p+,cur.r);
     }
     printf("%lld\n",cur.v);
     ;
 }