面试题 16.11. 跳水板
题目:
你正在使用一堆木板建造跳水板。有两种类型的木板,其中长度较短的木板长度为shorter,长度较长的木板长度为longer。你必须正好使用k块木板。编写一个方法,生成跳水板所有可能的长度。
返回的长度需要从小到大排列。
示例:
输入:
shorter = 1
longer = 2
k = 3
输出: {3,4,5,6}
提示:
0 < shorter <= longer
0 <= k <= 100000
分析:本题目特别要注意的两个点:
(1)当k=0时,无法组成跳水板,此时的输出为[]而不是[0],
(2)当shorter==longer时,此时无论怎么组合只能组成一种跳水板,避免重复。
解答:
1.先考虑正常情况,然后单独考虑k=0时的情况,最后通过将数组放入集合的方法去除重复,但是这里去除重复后顺序就乱了,要重新排序,这种方法比较耗时,因为要排序。
class Solution:
def divingBoard(self, shorter: int, longer: int, k: int) -> List[int]:
result = []
for i in range (k+1):
result.append((k-i)*shorter + i*longer)
if (result == [0]):
result = []
result = list(set(result))
result.sort()
return result
2.先考虑shorter和longer相等的情况,然后考虑正常情况,最后单独考虑k=0时的情况,这种方法避免了重复,相对于上面的第一种方法更简单一些。
class Solution:
def divingBoard(self, shorter: int, longer: int, k: int) -> List[int]:
result = []
if shorter == longer:
result.append(shorter * k)
else:
for i in range (k+1):
result.append((k-i)*shorter + i*longer)
if (result == [0]):
result = []
return result