新开了一个系列,主要记录在刷每日一题的时候学到的内容。
2020.03.27
题目猛以看还是挺唬人的,如果暴力破解复杂度
O
(
n
∗
n
)
O(n*n)
O(n∗n).
首先毫无疑问是要统计每个数字出现的次数的。
其次在知道了每个数字出现的次数以后,我们需要找到所有次数的最大公因数。
因此题目其实是可以当成两个子问题来解决的。
class Solution(object):
def hasGroupsSizeX(self, deck):
"""
:type deck: List[int]
:rtype: bool
"""
if not deck or len(deck)<2:
return False
# 统计数字出现次数
deck.sort()
i,j = 0 ,0
num = []
while i <len(deck):
j = i+1
while j < len(deck):
if deck[j] != deck[i]:
num.append(j-i)
break
j += 1
if j == len(deck):
num.append(j-i)
i = j
# 求解最大公因数 (考虑了最小的一个数字)
num.sort()
ans = []
for i in range(2,num[0]+1):
if num[0] % i == 0:
ans.append(i)
flag = 1
for k in ans:
flag = 0
for x in num:
if x%k != 0:
flag = 1
if flag == 0:
return True
return flag==0
上面的代码是我最初提交的版本,但是很明显有改进的空间。接下来就具体解析。
首先是计数,可以使用collections.Counter
vals = collections.Counter(deck).values()# 返回值是个列表
Counter
常用的接口:
Count(obj).values()
返回统计次数Count(obj).item()
返回统计的键值对列表Count(obj).most_common(n)
返回统计次数最多的键值对列表
其次是求解最大公约数gcd,可以用python的自带math.gcd(m, n)
也可以按照辗转相除法自己编写:
def gcd(m, n):
if n == 0:
return m
return gcd(m, n%m)
最后,如果要计算全部数据的最大公因数可以用functools.reduce
functools.reduce
的用法是对一列数组重复进行计算,直到返回值为只有一个
ret = reduce(lambda x,y: x+y,[1,2,3,4])
# 对列表持续加和,最后返回1+2+3+4=10
所以整个题目其实可以用一句话解决
return functools.reduce(math.gcd, collections.Counter(deck).values())>=2