C. Perform Operations to Maximize Score （cf div2）

C. Perform Operations to Maximize Score

大致题意：

有一个长度为n的数组，其中部分位置的值可以通过操作来加1（0-n个位置），最多操作k次。任取一个值a[i]，剩下的值取中位数med，然后两者之和为分数。求分数的最大值。

思路：

发现在进行k次操作并排序后，最大分数为$ans=a[n/2]+a[n]$

所以考虑如何操作使得ans最大：
只有两种操作思路，1.使a[n/2]最大 2.使a[n]最大
可以分别进行两种操作，最后令ans取max

1.二分答案找到最大可能的中位数
check方法：如果x可以是中位数，那么小于x的元素数量一定不到 $n/2$ 个，而且cost一定小于等于k。

2.使a[n]最大
易知当最大值a[n]可以操作时，将k全部加到最大值a[n]上，此时k全部有贡献，ans直接为最大ans。
否则将最大可操作的值加满。

代码:

#include <bits/stdc++.h>
#define endl '\n'
#define int long long
typedef long long ll;
using namespace std;

const int MAXN=2e5+5;

// 定义一个结构体来存储数组元素及其对应的二进制标记
struct Point {
	int x;  // 数组元素
	int bj; // 二进制标记，表示是否可以选择该元素进行增加操作
};

// 重载小于运算符，用于排序
bool operator<(const Point& a, const Point& b) {
	return a.x < b.x || (a.x == b.x && a.bj < b.bj);
}

void solve() {
	int n, k;
	cin >> n >> k; // 读取数组长度和操作次数
	vector<Point> a(n + 1); // 初始化结构体数组a
	
	// 读取数组元素和二进制标记
	for (int i = 1; i <= n; ++i) {
		cin >> a[i].x;
	}
	for (int i = 1; i <= n; ++i) {
		cin >> a[i].bj;
	}
	
	// 对数组进行排序
	sort(a.begin() + 1, a.end());
	
	// 初始答案为中间元素和最大元素的值之和
	int ans = a[n / 2].x + a[n].x;
	int tmp = 0;
	
	//如果最大值是直接可以操作的，直接全部加到最大值上，输出答案
	if(a[n].bj==1){
		cout<< ans + k <<endl;
		return;
	}
	
	// 查找可操作的最大元素
	for (int i = n; i >= 1; --i) {
		if (a[i].bj) {
			tmp = i;
			break;
		}
	}
	// 如果没有可操作的元素，直接输出初始答案
	if (tmp == 0) {
		cout << ans << endl;
		return;
	}
	
	// 1.二分查找最大可能的中位数
	int L = a[n / 2].x, R = L + k;
	while (L <= R) {
		int mid = (L + R) / 2;
		int all = n, cost = k;
		
		// 检查是否可以将足够的元素增加到 mid
		for (int i = n; i > 0 && all >= n / 2; --i) {
			if (a[i].x < mid && a[i].bj) {
				if (a[i].x + cost >= mid) {
					cost -= mid - a[i].x;
					--all;
				}
			}
			if (a[i].x >= mid) --all;
		}
		// 如果可以，则更新答案并尝试更大的中位数
		if (all < n / 2) {
			L = mid + 1;
			ans = max(ans, mid + a[n].x);
		} else {
			R = mid - 1;
		}
	}
	
	// 2.将最大可操作的数加k
	a[tmp].x+=k; 
	sort(a.begin() + 1, a.end());
	ans = max(ans,a[n / 2].x + a[n].x);
	
	// 输出答案
	cout << ans << endl;
}

signed main() {
	cin.tie(0)->ios::sync_with_stdio(0);
	int T = 1;
	cin >> T;
	while (T--) {
		solve();
	}
	return 0;
}