CF1832D2 Red-Blue Operations (Hard Version)
题目概述
给你 \(\{a_n\}\),第 \(i\) 次操作,如果是你第奇数次操作当前位置则令它 \(+i\) 否则 \(-i\)。
给出 \(q\) 个询问,问你进行完 \(k\) 个操作之后 \(a\) 中的最小值最大是多少?
题目分析
这道题目一眼二分答案。
先对 \(a\) 排序。
观察题目性质(经典):
- 对于当前位置 \(x\),如果我在 \(i\) 的时候操作了一次,那么我下一次一定会操作它,这样会使得代价最小(只减少 \(1\))。
因为我们考虑的是二分答案,所以说只需要确定对于每一个 \((k,val)\) 去检查 \(k\) 步能不能实现全部都大于等于 \(val\)。
首先确定 \(k\) 的下界。
对于第 \(i\) 个数,我们想要将它变到大于等于 \(x\),于是有:
\[a_i+k-i+1\geq x
\]
移项得到:
\[k\geq c+(i-1-a_i)
\]
为什么呢,因为我肯定是对于小的那个给予最大的关怀。
取前缀最大值即可。
这个我门直接找 \(<val\) 即可。
设 \(k=2w+p\),我们假设后面用 \(p\) 次加,那么显然,这 \(p\) 个可以加进去,这个我们可以假定 \(p\) 为 \(<val\) 的个数,然后如果奇偶性不同,则让 \(p+1\) 即可。
如果说 \(k\) 比下界小,那就不行,如果说 \(p\) 进行 \(+1\) 之后若 \(>n\) 就不行,因为此时判的情况就是 \(p=n\) 时但奇偶性不同所以说就会再减了一次。
还需要判断 \(k\) 的上界。
显然只需要满足:
\[sum-n\times val+(k+k-1+\dots+k-p+1)>w
\]
其意义时总和减去下面都有的部分,再加上一开始加的部分,这些部分就是我要可能减去的部分,如果这个部分不够减,那不好意思,就不行。
然后就能 AC 了。
代码
时间复杂度 \(\mathcal{O}(q\log V\log n)\)。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <stdlib.h>
#include <algorithm>
#include <vector>
#define int long long
#define N 200005
using namespace std;
int n,q,sum;
int a[N],lim[N];
bool check(int k,int c) {int p = lower_bound(a + 1,a + n + 1,c) - a - 1;if (p == 0) return true;if (k < p) return false;if ((k - p) & 1) p ++;if (p > n) return false;if (k < c + lim[p]) return false;if (sum * 2 - n * c * 2 + p * (2 * k - p + 1) >= k - p) return true;return false;
}
signed main(){cin >> n >> q;for (int i = 1;i <= n;i ++) scanf("%lld",&a[i]),sum += a[i];stable_sort(a + 1,a + 1 + n);for (int i = 1;i <= n;i ++) lim[i] = i - a[i] - 1;for (int i = 2;i <= n;i ++) lim[i] = max(lim[i],lim[i - 1]);for (int i = 1;i <= q;i ++) {int k;scanf("%lld",&k);int l = 0,r = 1e18,res = 0;while(l <= r) {int mid = l + r >> 1;if (check(k,mid)) res = mid,l = mid + 1;else r = mid - 1;}printf("%lld ",res);}return 0;
}