Red-Blue Operations (Hard Version)

Red-Blue Operations (Hard Version)

题意：给一个数组，有q组操作，每组操作ki次，问操作完后数组中最小数最大是多少，n，q<=2e5。每次操作过程如下：

设当前操作次数为cnt；
选一个数，如果它是红色，则+cnt，如果是蓝色，则-cnt，
然后cnt++，如果当前是红色，变成蓝色，否则变成红色。

思路：

如果k小于n就直接加，如果k大于n就把加放在后面，后面全部加，前面的分组每一组为+cnt，-（cnt+1）如果k-n是偶数，那就要减（k-n）/2，如果是奇数，就减（k-n+1）/2。
具体看代码。

代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#define int long long
#define endl '\n'

using namespace std;

const int maxn = 3e5 + 7;
const int inf = 1e18;

int n, q, a[maxn];

signed main(){
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	cin >> n >> q;
	int sum = 0;
	for(int i = 1; i <= n; i ++ )	cin >> a[i], sum += a[i];
	sort(a + 1, a + n + 1);
	vector<int> f(n + 2, inf);
	for(int i = 1; i <= n; i ++ )
		f[i + 1] = min(f[i], a[i]) + 1;
	while(q -- ){
		int k;
		cin >> k;
		if(k < n){
			cout << min(f[k + 1], a[k + 1]) << ' ';
		}
		else {
			int ans = 0;
			int s = sum;
			if((k - n) % 2 == 0){
				ans = f[n + 1] + k - n;
				s += n * (k + k - n + 1) / 2 - (k - n) / 2;
			}
			else {
				ans = min(f[n] + k - n + 1, a[n]);
				s += (n - 1) * (k + k - n + 2) / 2 - (k - n + 1) / 2;
			}
			cout << min(ans, s / n) << ' ';
		}
	}
	cout << endl;
}