D1. Red-Blue Operations (Easy Version)（贪心/思维）

题目
参考

题意

给定长度为n(1<=n<=1000)的数组
1<=a[i]<=10^9
q个查询(1<=q<=1000)，每次查询，给定整数k(1<=k<=10^9)

初始时，每个元素状态都是红色。
对于第i次操作，每次选择从数组中选择一个元素进行操作。

• 如果当前元素状态为红色，则将该元素的值+i，并变更当前元素状态为蓝色。
• 如果当前元素状态为蓝色，则将该元素的值-i，并变更当前元素状态为红色。

问，经过k次操作后。得到的新数组，它的元素最小值min，最大能取多大。即求数组的最大 最小值。

思路

当k<=n时，我们贪心地把每个操作，应用到最小的k个元素，即可满足。

当k>n时，复用贪心思路。这里先假设n-k是偶数。

我们把前n大的元素，累加，应用到这n个元素。
怎么确保到最后n次操作时，每个元素的状态刚好是红色（即对应新增状态）呢？我们只要保证，在最后n次操作之前，每个元素都应用了偶数次即可。

为了确保在应用最后n次操作的时候，元素之间的变更最少（这里需要理解下，因为我们要求的是数组的最小值，因此，需要确保整体增加的时候，是相对均匀的）。
我们可以把前面k-n次操作，拆分成(+1,-2)，(+3,-4)…这样的成对操作，每个操作都只-1。

设a’为应用了最后n次操作的新数组。设num为剩余的-1操作。
令res = a’[i]-min(a’)。那么这res可以抵消掉数组的-1操作，且不会变更数组最小值。

经过res次-1操作后，数组a’则变成所有元素都相等了，那么之后，每n次-1，我们的数组最小值都会-1。（这里注意向上取整）

详见代码。

代码

#include<string>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
#define ll long long
const int maxn = 1010;

int n, q, k;
int a[maxn], b[maxn];

int cal() {
	scanf("%d", &k);
	for (int i = 0; i < n; ++i) {
		b[i] = a[i];
	}
	if (k <= n) { // add [k,...,1] to minist topk elems.
		for (int i = 0; i < k; ++i) {
			b[i] += k - i;
		}
		sort(b, b + n);
		return b[0];
	}
	
	int num = k - (n - 1); // 用于统计操作数剩余量 
	for (int i = 0; i < n - 1; ++i) {
		b[i] += k - i;
	}
	if ((k - n) % 2 == 0) { // n 和 k 奇偶性相同 
		b[n-1] += k - (n - 1);
		--num;
	}
	sort(b, b + n);
	num /= 2; // (increase,decrease)的对数 
	
	int tmp = 0; // 逐个减去，不要加起来，防爆long long 
	for (int i = 1; i < n; ++i) {
		tmp = b[i] - b[0];
		num = max(0, num - tmp);
	} 
	
	num = (num + n - 1) / n; // mininum 需要 -1 的次数 ，需要向上取整 
	
	return b[0] - num;
}
void solve() {
	scanf("%d%d", &n, &q);
	for (int i = 0; i < n; ++i) {
		scanf("%d", &a[i]);
	}
	sort(a, a + n);
	while (q--) {
		printf("%d%c", cal(), !q ? '\n' : ' ');
	}
} 
int main() {
	int t = 1;
//	scanf("%d", &t);
	while (t--) {
		solve();
	}
}
/*

*/

同名公众号

对方正在debug