3578 最大中位数（二分查找）

1. 问题描述：

给定一个由 n 个整数组成的数组 a，其中 n 为奇数。你可以对其进行以下操作：
选择数组中的一个元素（例如 ai），将其增加 1（即，将其替换为 ai+1）。
你最多可以进行 k 次操作，并希望该数组的中位数能够尽可能大。奇数长度的数组的中位数是数组以非降序排序后的中间元素。例如，数组 [1，5，2，3，5] 的中位数为 3。

输入格式

第一行包含两个整数 n 和 k。第二行包含 n 个整数 a1，a2，…，an。

输出格式

输出一个整数，表示通过操作可能得到的最大中位数。

数据范围

对于 30% 的数据，1 ≤ n ≤ 5。
对于 100% 的数据，1 ≤ n ≤ 2 × 10 ^ 5，1 ≤ k ≤ 10 ^ 9，1 ≤ ai ≤ 10 ^ 9。

输入样例1：

3 2
1 3 5

输出样例1：

5

输入样例2：

5 5
1 2 1 1 1

输出样例2：

3

输入样例3：

7 7
4 1 2 4 3 4 4

输出样例3：

5
来源：https://www.acwing.com/problem/content/3581/

2. 思路分析：

分析题目可以知道题目是具有单调性的（看到题目数据范围为10 ^ 5而且在某个限制条件下的最值问题需要考虑能否使用二分来解决），当中位数越大的时候那么需要的操作次数是越多的，如果当前枚举的答案为ans，ans左边的数字都是满足题目要求的，ans右边的数字是不满足要求的，所以我们可以使用二分来解决；我们可以枚举答案，也即枚举中位数，枚举的区间范围为[1，2e9]，判断当前数组a的中位数变为当前枚举的mid是否可以在小于等于k次操作内完成，如果可以说明中位数大于等于当前枚举的mid，否则中位数一定小于mid，根据check函数的结果来更新下一次二分的区间范围。

3. 代码如下：

from typing import List


class Solution:
    # 判断将a的中位数变为x的操作次数是否小于等于k, 计算中间位置到a的末尾的位置变为x需要的操作次数count
    def check(self, x: int, k: int, a: List[int]):
        count = 0
        for i in range(len(a) // 2, len(a)):
            if a[i] < x:
                count += x - a[i]
        return count <= k

    def process(self):
        n, k = map(int, input().split())
        a = list(map(int, input().split()))
        # 从小到大排序
        a.sort()
        # 注意中位数可能是1e9, 所以r要设置为2e9
        res = 0
        l, r = 1, 2 * 10 ** 9
        while l <= r:
            mid = l + r >> 1
            if self.check(mid, k, a):
                res = mid
                l = mid + 1
            else:
                r = mid - 1
        return res


if __name__ == '__main__':
    print(Solution().process())