印象最深的一题应该是这个:给定一个非负整数数组和一个整数 m,你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小。
主要思想是二分查找
题目要求的是分割成连续的子数组,并且结果为所有满足分组个数的最大和的最小值。仔细想想 最后的结果应该是介于原数组的最大值和原数组的和之间的某个值。
所以用二分查找。
low = max(nums) high = sum(nums) mid = (low+high) // 2
相当于向下取整。
假设此时mid
就是最后的结果,这时分割的子数组的个数num
与要求的m
进行比较。
- 如果
num > m
,说明此时的mid
比正确结果小了,因此应该在右区间继续查找; - 如果
num < m
,说明此时的mid
比正确结果大了,因此应该在左区间继续查找。
还有一种情况,如果分割的子数组的个数num == m
,此时也不一定是正确答案。正如示例给的那样,当num == m
时有4种分割方法。
原数组为 [7,2,5,10,8]
法1:[7] , [2,5,10,8]
结果为25
法2:[7,2] , [5,10,8]
结果为23
法3:[7,2,5]] , [10,8]
结果为18
法4:[7,2,5,10] , [8]
结果为24
因此,此时虽然num = m
,只要low < high
,就应该在右区间中继续找。
代码如下,不明白的可以手推一遍:
class Solution(object):
def splitArray(self, nums, m):
"""
:type nums: List[int]
:type m: int
:rtype: int
"""
left,right = max(nums),sum(nums)
def test_mid(mid):
num = 1
s = 0
for i in nums:
if s+i >mid:
num +=1
s = i
else:
s += i
return num >m
while left<right:
mid = (left+right) //2
b = test_mid(mid)
if b:
left = mid+1
else:
right = mid
return left
第2题:给定一个字符串,请将字符串里的字符按照出现的频率降序排列。
这里我能想到用dict
字典来解决,只不过对排序不太了解,想要了解sort函数和sorted函数的可以参考在Python中如何使用sorted()和sort()函数
通过sorted的方法进行value排序,对字典排序后无法直接按照字典进行返回,返回的为列表元组:
对value值进行从大到小排序:
s = sorted(s_dict.items(), key=lambda item:item[1], reverse = True)
对key进行从大到小排序:
s = sorted(s_dict.items(), key=lambda item:item[0])
代码如下:
class Solution(object):
def frequencySort(self, s):
"""
:type s: str
:rtype: str
"""
from collections import Counter
s_dict =Counter(s)
print(s_dict)
s_sort = sorted(s_dict.items(),key=lambda item:item[1],reverse=True)
print(s_sort)
res = ""
for key,value in s_sort:
res +=key*value
return res
第3题:**编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为:对于一个正整数,每一次将该数替换为它每个位置上的数字的平方和,然后重复这个过程直到这个数变为 1,也可能是 无限循环 但始终变不到 1。如果 可以变为 1,那么这个数就是快乐数。
如果 n 是快乐数就返回 True ;不是,则返回 False
主要一点:如果结果出现重复的则返回false,这里用到了一个set()
代码如下:
class Solution(object):
def isHappy(self, n):
"""
:type n: int
:rtype: bool
"""
mm = set()
while n!=1:
summ = 0
while n>0:
tmp = n%10
summ += tmp**2
n //=10
if summ in mm:
return False
else:
mm.add(summ)
n = summ
return True