C. Robin Hood in Town思考与理解

文章目录

C. Robin Hood in Town


  • 首先就是得思考一个问题:如何快速找到有多少个数严格小于平均数的一半?答案是显而易见的,二分,所以我们需要进行升序排序
  • 考虑题目给出的特殊情况当n=1或n=2的时候,直接输出-1即可
  • 那么对于可以通过增加x来判断满足最小的x的问题,这里就转换为,如何快速增加x,然后逐一判断这个x所带来的影响?
    • 直接暴力的做法肯定是x从0开始,逐个+1显而易见,这样十分慢
    • 正确的做法是直接使用二分,这里就要考虑这个二分的范围了,最小肯定是0,那么最大是多少?假设n全部放在一个测试用例,那么考虑到最大的数是10**6,n的最大范围2*10**5,如果开始的全部的n都是1,那么我们得将一半的数全部变为最大也就是10**11
python 复制代码
import bisect
# 二分+二分的问题
t = int(input())
for _ in range(t):
    n = int(input())
    a = list(map(int, input().split()))
    if n == 1 or n == 2:
        print(-1)
        continue
    a.sort()
    asum = sum(a)
    # aver = asum / n
    # index = bisect.bisect_left(a, aver/2)
    # if index > n // 2:
    #     print(0)
    #     continue
    # 接下来怎么办?
    # 增加数的问题,具有二分的性质
    # 考虑增加的数量为mid
    def check(mid):
        a[-1] += mid
        tmpsum = asum + mid
        aver = tmpsum / n
        index1 = bisect.bisect_left(a, aver/2)
        a[-1] -= mid
        return index1 > n // 2
    l,r = 0,10**12
    res = float('inf')
    while l<=r:
        mid = (l+r)//2
        if check(mid):
            res = min(res,mid)
            r = mid - 1
        else:
            l = mid + 1
    print(res)
相关推荐
PAK向日葵1 小时前
【算法导论】如何攻克一道Hard难度的LeetCode题?以「寻找两个正序数组的中位数」为例
c++·算法·面试
爱喝矿泉水的猛男3 小时前
非定长滑动窗口(持续更新)
算法·leetcode·职场和发展
YuTaoShao3 小时前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
YouQian7724 小时前
Traffic Lights set的使用
算法
go54631584655 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法
aramae5 小时前
大话数据结构之<队列>
c语言·开发语言·数据结构·算法
大锦终5 小时前
【算法】前缀和经典例题
算法·leetcode
想变成树袋熊6 小时前
【自用】NLP算法面经(6)
人工智能·算法·自然语言处理
cccc来财6 小时前
Java实现大根堆与小根堆详解
数据结构·算法·leetcode
Coovally AI模型快速验证7 小时前
数据集分享 | 智慧农业实战数据集精选
人工智能·算法·目标检测·机器学习·计算机视觉·目标跟踪·无人机