标签：le int 超级 pos 钢琴 ch NOI2010 sec sum
知识点：RMQ，堆，小技巧
有趣的套路。
简述
给定一长度为 \(n\) 的数列 \(a\)，给定参数 \(k,L,R\)，求最大的 \(k\) 个长度在 \([L,R]\) 间的子段之和。
\(1\le n,k\le 5\times 10^5\)，\(0\le |A_i|\le 10^3\)，\(1\le L\le R\le n\)，保证有解。
2S，512MB。
分析
一种显然的暴力是枚举所有长度在 \([L,R]\) 内的区间，大力求和并放入大根堆中，取最优的前 \(k\) 个即可。复杂度 \(O\left(\left(n^2 + k\right)\log n^2\right)\) 级别。期望得分 20pts。
而 \(n,k\) 同阶，上述做法的问题主要在于堆中有许多无贡献的元素。考虑如何缩小堆的大小，但又能保证每次访问堆顶时取出的元素是最优的。本题提供了一种解决此类前 \(k\) 优问题的思路。
首先求得 \(a\) 的前缀和 \(\operatorname{sum}\)，任一区间都可以表示成两个前缀相减的形式，则题目所求即为最大的 \(k\) 个 \(\operatorname{sum}_r - \operatorname{sum}_{l-1}\) 的值。若左端点 \(l-1\) 固定，最优的右端点 \(r\) 可以通过对 \(\operatorname{sum}\) 建 ST 表查询区间 \([l-1 + L, l-1 + R]\) 的最大值 \(O(1)\) 求得。
对于上述过程，定义状态 \((v, l-1, pos, x, y)\) 表示左端点为 \(l-1\)，查询区间为 \([x, y]\)，在此区间内找到的最优的 \(r\) 为 \(pos\)，\(\operatorname{sum}_r - \operatorname{sum}_{l-1}\) 的值是 \(v\)。
初始时枚举所有左端点 \(l-1\)，按照上述 ST 表做法构造初始状态 \((v, l - 1, pos, l - 1 +L, i - 1 + R)\) 并放入以状态中 \(v\) 为关键字的大根堆中。每次取出堆顶的状态，统计其贡献 \(v\)，然后构造新状态 \((v', l - 1, pos', x, pos - 1)\) 与 \((v'', l - 1, pos'', pos +1, y)\) 放入堆中。可以发现这样能够保证枚举到所有有贡献的状态。
堆中初始有 \(n\) 个状态，每次查询分裂状态都会使堆的大小 \(+1\)，总复杂度 \(O((n+k)\log (n +k))\) 级别。
实现时注意一些边界细节。
代码
//知识点：RMQ，堆，小技巧
/*
By:Luckyblock
*/
#include <algorithm>
#include <cctype>
#include <cstdio>
#include <cstring>
#include <queue>
#define pr std::pair
#define mp std::make_pair
#define LL long long
const int kN = 5e5 + 10;
//=============================================================
struct Data {
LL val;
int i, pos, l, r;
bool operator < (const Data &sec_) const {
return val < sec_.val;
}
};
int n, k, l, r;
LL ans, sum[kN];
std::priority_queue <Data> q;
//=============================================================
inline int read() {
int f = 1, w = 0;
char ch = getchar();
for (; !isdigit(ch); ch = getchar())
if (ch == '-') f = -1;
for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0');
return f * w;
}
void Chkmax(int &fir, int sec) {
if (sec > fir) fir = sec;
}
void Chkmin(int &fir, int sec) {
if (sec < fir) fir = sec;
}
namespace ST {
const int kMaxn = kN;
const int kMaxLog = 20 + 1;
LL mx[kMaxn][kMaxLog];
int Log2[kMaxn], pos[kMaxn][kMaxLog];
void Build() {
for (int i = 0; i <= n; ++ i) {
if (i > 1) Log2[i] = Log2[i >> 1] + 1;
mx[i][0] = sum[i];
pos[i][0] = i;
}
for (int i = 1; i <= kMaxLog; ++ i) {
for (int j = 0; j + (1 << i) - 1 <= n; ++ j) {
if (mx[j][i - 1] >= mx[j + (1 << (i - 1))][i - 1]) {
mx[j][i] = mx[j][i - 1];
pos[j][i] = pos[j][i - 1];
} else {
mx[j][i] = mx[j + (1 << (i - 1))][i - 1];
pos[j][i] = pos[j + (1 << (i - 1))][i - 1];
}
}
}
}
int Query(int L_, int R_) {
int lth = Log2[R_ - L_ + 1];
return mx[L_][lth] > mx[R_ - (1 << lth) + 1][lth] ?
pos[L_][lth]:
pos[R_ - (1 << lth) + 1][lth];
}
}
//=============================================================
int main() {
n = read(), k = read(), l = read(), r = read();
for (int i = 1; i <= n; ++ i) sum[i] = sum[i - 1] + read();
ST::Build();
for (int i = 0; i <= n; ++ i) {
if (i + l > n) break; //无贡献
int pos = ST::Query(i + l, std::min(n, i + r));
q.push((Data) {sum[pos] - sum[i], i, pos, i + l, std::min(n, i + r)}); //注意取 min
}
while (k --) {
Data t = q.top(); q.pop();
ans += t.val;
int i = t.i, pos = t.pos, l_ = t.l, r_ = t.r;
int new_pos = ST::Query(l_, pos - 1);
if (l_ <= pos - 1) q.push((Data) {sum[new_pos] - sum[i], i, new_pos, l_, pos - 1});
new_pos = ST::Query(pos + 1, r_);
if (pos + 1 <= r_) q.push((Data) {sum[new_pos] - sum[i], i, new_pos, pos + 1, r_});
}
printf("%lld\n", ans);
return 0;
}
标签：le,int,超级,pos,钢琴,ch,NOI2010,sec,sum 来源： https://www.cnblogs.com/luckyblock/p/14333221.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。