首先看题目:
你正在使用一堆木板建造跳水板。有两种类型的木板,其中长度较短的木板长度为shorter,长度较长的木板长度为longer。你必须正好使用k块木板。编写一个方法,生成跳水板所有可能的长度。
返回的长度需要从小到大排列。
示例:
输入:
shorter = 1
longer = 2
k = 3
输出: {3,4,5,6}
提示:
0 < shorter <= longer
0 <= k <= 100000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/diving-board-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
---------------------------------------------------------------------------
该题目属于简单难度。刚刚看到该题时,首先想到的是递归做法,然后,果不其然的超时了。。。。如果想要使用递归做法的话,就需要新建一个数组,空间换时间才能保证不超时。但是比较复杂,而且时间复杂度和空间复杂度都为O(n)了,对于这个题来说不算低了。
经过思考后,发现:
因为给定了shorter值以及longer值,并且每次只能由这两种之一来拼凑跳水板,所以很明显可以找出k块跳水板的长度的上下界:
min = k * shorter。
max = k * longer。
在确定上下界后,进行min - > max的遍历,中间的步骤就是不断将原先长度为shorter的木板更换为长度为longer的木板的过程。整个过程就是一个排列组合的问题。
替换的过程用代码表示为:
for(int i = min;i <= max;i = i+longer-shorter){
ans[index++] = i;
}
-----------------------------------------------------------------------
上实现代码:
JAVA版(执行用时与内存消耗均击败100%用户23333333):
class Solution {
public int[] divingBoard(int shorter, int longer, int k) {
if(k == 0) return new int[0];
if(longer == shorter) return new int[]{k*longer};
int []ans = new int[k+1];
int index = 0;
int min = k*shorter;
int max = k*longer;
//解法二:这种线性复杂度。
for(int i = min;i <= max;i = i+longer-shorter){
ans[index++] = i;
}
// 解法一 。这种会超时。
// solve(ans, shorter, longer, k, 0, 0, 0);
return ans;
}
// dp[i] = dp[i-1] + shortes || dp[i-1] + longest
public void solve(int []ans, int shorter, int longer, int k, int index, int tk, int sum){
if(tk == k){
//到当前为止,以及拼了tk段了
ans[index] = sum;
return;
}
solve(ans, shorter, longer, k, index++, tk+1, sum+shorter);
solve(ans, shorter, longer, k, index++, tk+1, sum+longer);
}
}
C++版(执行用时击败60%,内存消耗击败100%。。):
class Solution {
public:
vector<int> divingBoard(int shorter, int longer, int k) {
vector<int> ans;
if(k == 0) return ans;
if(longer == shorter){
ans.push_back(k*longer);
return ans;
}
int min = k * shorter;
int max = k * longer;
for(int i = min;i <= max;i = i + longer - shorter){
ans.push_back(i);
}
return ans;
}
};
python版(python写算法还是不大习惯,,,爱加上分号、定义类型《捂脸》。执行用时和内存消耗击败比和用C++差不多。。。):
class Solution:
def divingBoard(self, shorter: int, longer: int, k: int) -> List[int]:
ans = []
if(k == 0):
return ans
if(shorter == longer):
ans.append(k*longer)
return ans
min = k * shorter
max = k * longer
# index = 0
while(min <= max):
ans.append(min)
# index = index + 1
min = min + longer - shorter
return ans