蓝桥杯3527阶乘的和 | 组合数学

写在前面的话

昨天的蓝桥杯每日一题 奇怪的数 用python写太难了，甚至题解的优化处理也只能得70分（后几个TLE了），但是C++写 不优化的6层for循环+后4位复用 可以直接过。（雾）

题目传送门

---

这个题目的思路是对A_i排序后，记录每个A_i出现的次数dic[A_i]。显然最小的A₀满足m。每次遍历dic，如果m对应的A_i出现的次数是A_i+1的倍数，那么将其并入dic[A_i+1]，同时m++。当m无法满足次数的倍数关系时m值即为所求。

---

from collections import defaultdict
n = int(input())
ai = list(map(int, input().split()))
ai.sort()
m = ai[0]
dic = defaultdict(int)	# 因为python的dict()不支持类似C++中map的运算操作，所以引入defaultdict

for it in ai:
    dic[it] += 1

while True:
    x = dic[m]
    if x %(m + 1) != 0:
        break
    dic[m+1] += x//(m+1)
    m += 1

print(m)

END✨

---