蓝桥杯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值即为所求。

由于普通字典不能对未使用的键赋值，所以调用defaultdict。

---

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)

---

后记
在《蓝桥杯算法入门》中提供了可以不调defaultdict使用普通字典的的处理方法，即每次对不在当前字典的键赋值为1，实现如下：

for i in a:
    x = min(i,x)
    if i not in dic:	dic[i] = 1
    else :	dic[i] += 1

END✨

---