链接:
题目大意:
题面很长,说人话就是,给一个长n的序列a,另外给出k,m,c,d四个整数,我们可以选择序列中的一个位置p(1 ≤ p ≤ n - m + 1),并对于所有0 ≤ i < m,将ap+i增加(c + di),求至多进行一次操作后,序列中第k大的值最大可能为多少。
数据范围:1 ≤ k, m ≤ n ≤ 2 × 105,0 ≤ c, d ≤ 109,0 ≤ ai ≤ 109。
输入格式:第一行五个整数:n, k, m, c, d;第二行n个整数ai。
分析思路:
说实话,看到第k大,俺就想平衡树,结果呢,是个二分…
首先,答案是唯一确定的,并且根据数据范围一定属于[0, 1018] (其实应该比1018要小,但这样更保险),我们二分的根据就是可以讲这个区间划分为小于等于答案和大于答案两部分。
从时间复杂度上我们不难看出,如果写二分的话,那么check中的时间复杂度应该是O(n)的,显然很难想…
因为这个A搞了个差分,所以就不免往差分和前缀和这里去想了,结果还真能做;首先check返回true时表示传入的mid小于等于答案,即在最优的位置加上这个等差数列之后,大于等于mid的数的个数大于等于k,反之则为false;扫描一遍序列a,如果ai≥mid则cnt++,反之,尝试寻找一段有效区间(指首项允许加在的位置上)使得首项加在该区间任意位置后,可以使得该项大于等于mid,而记录这个区间,我们采用差分的方式,即构造一个差分数组s,令区间[l, r]的左s[l]++,右s[r + 1]–,这样再重新扫描一遍就可以得到,当等差数列加在最优位置后,可以多出多少个大于mid的数,再加上原来的cnt,最后与k比较,这样就线性的完成了这项工作。
然后这个题就可以A掉了吧…
代码如下:
#include<iostream>
#include<cstring>
#include<cmath>
typedef long long ll;
const int N = 200010;
const ll MAX = 1e18;
int n, k, m;
ll c, d;
ll a[N];
int s[N];
bool check(ll x){
for (int i = 0; i <= n + 1; i ++ )
s[i] = 0;
int cnt = 0, l, r;
for (int i = 1; i <= n; i ++ )
if (a[i] >= x)
cnt++;
else{
//首项最远的位置
l = i - m + 1;
//寻找右边界
if (d != 0)
r = std::min(int(i - ceil(double(x - a[i] - c) / d)), i);
else if (a[i] + c >= x)
r = i;
else
r = -0x3f3f3f3f;
//如果区间存在则记录,如果l>r表示数列最大项加上也不够
if (l <= r)
s[std::max(1, l)]++, s[std::max(0, r) + 1]--;
}
for (int i = 1; i <= n; i ++ ){
s[i] += s[i - 1];
if (s[i] + cnt >= k)
return true;
}
return false;
}
int main(){
scanf("%d%d%d%lld%lld", &n, &k, &m, &c, &d);
for (int i = 1; i <= n; i ++ )
scanf("%lld", &a[i]);
ll l = 0, r = MAX, mid;
while (l < r){
mid = l + r + 1 >> 1;
if (check(mid))
l = mid;
else
r = mid - 1;
}
printf("%lld\n", l);
}