你正在使用一堆木板建造跳水板。有两种类型的木板,其中长度较短的木板长度为shorter,长度较长的木板长度为longer。你必须正好使用k块木板。编写一个方法,生成跳水板所有可能的长度。
返回的长度需要从小到大排列。
示例:
输入:
shorter = 1
longer = 2
k = 3
输出: {3,4,5,6}
提示:
由于需要K个木板,首先需要排除特殊情况:
然后,由于有shorter 和longer两个值,因此每一个值范围为0~k,则返回数组长度为k+1,一次遍历即可,每次的值为 shorter * i + (k - i) * longer;
最后,题目要求返回结果需从小到大排序,则调用Arrays.sort(ret)即可。
class Solution {
public int[] divingBoard(int shorter, int longer, int k) {
//step1:判断特殊情况
if(k == 0) return new int[]{};
if(shorter == longer) return new int[]{ shorter * k};
//Step2:定义结果变量
int[] ret = new int[ k + 1];
//Step3:一次遍历
for(int i = 0; i <= k; i++){
ret[i] = i * shorter + (k - i) * longer;
}
//Step4:排序
Arrays.sort(ret);
//Step5:返回结果
return ret;
}
}
如何降低执行时间:
每次都需要执行i * shorter + (k - 1) * longer,四次运算,通过观察发现,i递增时,都只增加了(longer - shorter),可以提前定义一个变量 diff = longer - shorter ,每次当I 递增时,都用ret[I - 1] + diff,只需要两次加法操作,整体运行时间提高了3ms.
int diff = longer - shorter;
ret[0] = k * shorter;
//Step3:一次遍历
for(int i = 1; i <= k; i++){
ret[i] = ret[i - 1] + diff;
}