问题描述
小蓝老师教的编程课有 N 名学生, 编号依次是 1…N 。第 i 号学生这学期刷题的数量是 Ai 。
对于每一名学生, 请你计算他至少还要再刷多少道题, 才能使得全班刷题比他多的学生数不超过刷题比他少的学生数。
输入格式
第一行包含一个正整数 N 。
第二行包含 N 个整数:A1,A2,A3,…,AN.
输出格式
输出 N 个整数, 依次表示第 1…N 号学生分别至少还要再刷多少道题。
样例输入
5 12 10 15 20 6
样例输出
0 3 0 0 7
评测用例规模与约定
对于 30% 的数据,1≤N≤1000,0≤Ai≤1000.
对于 100% 的数据, 1≤N≤100000,0≤Ai≤100000.
给出一个序列表示每个人目前的刷题数,计算每个人最少再刷多少题,可使得刷题数比他多的人数不超过刷题数比他少的人数。
给出一个数字,计算出序列中比它大的数字个数和比它小的数字个数
解题思路一:二分 算法复杂度:
import bisect
n = int(input())
a = list(map(int, input().split()))
sa = sorted(a) #获取一份和a相同的,但经过排序的列表
# #判断刷题数为cur时,刷题数比他多的人数是否不超过比他小的人数
def check(cur, raw):
few = bisect.bisect_left(sa, cur) #使用二分的方式,通过查找元素位置,计算数目
fewer = few if cur == raw else few - 1 #当cur的值相较原本已经增加,这时判断少刷题人数要扣掉自己
more = len(sa) - bisect.bisect_right(sa, cur) #计算多刷题的人数
if more <= fewer: #判断情况
return True
else:
return False
max_num = sa[-1] #获取所有人中刷题最多的题数,作为二分的右边界
res = [] #记录每人为实现目标要多刷的题数
for i in range(n):
l, r = a[i] - 1, max_num + 1 #l,r是新的刷题数,标定左右边界
while l + 1 != r: #二分算法
mid = l + r >> 1
if not check(mid, a[i]): #刷题数不达标,说明当前题数还较少,压缩左边界
l = mid
else:
r = mid #否则压缩右边界
res.append(r - a[i]) #最终符合要求的结果在r所指范围中,用r减去a[i]即可得到需要多刷的题数
print(*res)
使用前缀和,避免重复计算 算法复杂度:O(nlogn)
示例:5个学生,刷题数分别是12,10,8,15,6
1、建一个列表cnt统计每个刷题数对应的学生数
2、计算该列表的前缀和列表scnt,特别赋值scnt[0]=cnt[0],因为本题Ai是从0开始的,但其实前缀和的0一般不用,是从1开始的。
例如:刷题数为10的人,比他多的人more可以用scnt[M] - scnt[cur]=5-3=2,比他少的人fewer可以用前一个前缀和scnt[cur-1] = scnt[cur] - cnt[cur]=3-1=2,如果是通过再刷题达到的刷题数cur,那么fewer需要再-1,减掉一个再刷题前的自己。(因为需要再刷题说明肯定比之前刷题数多)
n = int(input())
a = list(map(int, input().split()))
N = 100010 # 稍微比范围大一点
cnt = [0] * N; scnt = [0] * N # 最后多的0没有影响
for i in range(n):
cnt[a[i]] += 1 # 统计刷题数为ai的人数
scnt[0] = cnt[0] # 特别赋值
for i in range(1, N):
scnt[i] = scnt[i - 1] + cnt[i] # 刷题数i的前缀和
def check(x, old):
more = scnt[N - 1] - scnt[x] # more:所有人数 - 刷题数x的前缀和
fewer = scnt[x] - cnt[x] # fewer:刷题数x的前缀和 - 刷题数x的人数
if x != old: # 需要达到刷题数和原来刷题数不一样
fewer -= 1 # 减掉一个原来的自己
if more <= fewer:
return True
else:
return False
res = [] # 记录每人为实现目标要多刷的题数
for i in range(n):
l, r = a[i] - 1, 100001
while l + 1 != r:
mid = l + r >> 1 # 右移1等价于//2
if check(mid, a[i]):
r = mid
else:
l = mid
res.append(r - a[i]) # 需要多刷的题数 = 最少需要达到的刷题数r - 原来刷题数a[i]
print(*res) # 加个*,可以打印出列表的每个元素,用空格隔开
通过前面的做法,我们发现,对于需要多刷题的人,他们最后都会刷到同一个题目数。例如样例中他们最后都会刷到同一个题目数13.
因此,我们选出一个最小的当前刷题数,用一次二分查找即可确定将要刷到的题数,记为pos
之后遍历a中所有数字,check()不需要再二分查找pos,如果满足check(),就说明新刷题数是0,否则新刷题数是pos - a[i]
算法复杂度:O(n)
n = int(input())
a = list(map(int, input().split()))
t = min(a) # 选出刷题数最少的,后面进行一次二分查找
N = 100010
cnt = [0] * N; scnt = [0] * N
for i in range(n):
cnt[a[i]] += 1
scnt[0] = cnt[0]
for i in range(1, N):
scnt[i] = scnt[i - 1] + cnt[i]
def check(x, old):
more = scnt[N - 1] - scnt[x]
fewer = scnt[x] - cnt[x]
if x != old:
fewer -= 1
if more <= fewer:
return True
else:
return False
# 没有for循环,只需要做一次二分
l, r = t - 1, 100001
while l + 1 != r:
mid = l + r >> 1
if check(mid, t):
r = mid
else:
l = mid
pos = r # 最少需要达到的刷题数pos
res = []
for i in range(n):
if check(a[i], a[i]): # 已经满足more <= fewer
res.append(0) # 不需要刷题
else: # 不满足more <= fewer
res.append(pos - a[i]) # 需要多刷的题数 = 最少需要达到的刷题数 - 原来刷题数
print(*res)