题目
你正在使用一堆木板建造跳水板。有两种类型的木板,其中长度较短的木板长度为shorter
,长度较长的木板长度为longer
。你必须正好使用k
块木板。编写一个方法,生成跳水板所有可能的长度。
返回的长度需要从小到大排列。
示例:
输入:
shorter = 1
longer = 2
k = 3
输出: {3,4,5,6}
提示:
- 0 < shorter <= longer
- 0 <= k <= 100000
解题思路
首先考虑两种边界情况。
1)如果 k=0,则不能建造任何跳水板,因此返回空数组。
2)如果 shorter和 longer 相等,则建造的跳水板的长度是唯一的,都等于 shorter∗k,因此返回长度为 1 的数组,数组中的元素为shorter∗k。
3)再考虑一般情况,即 shorter<longer 且 k>0。由于短木板和长木板一共使用 k 块,因此一共有 k+1 种组合,每种组合下建造的跳水板长度都是不一样的,一共有 k+1 种不同的长度。
证明:为什么会有 k+1 种不同组合?
不妨考虑以下两种组合
1)第一种组合,有i块长木板,则短木板有k-i块,总长度为 len1 = shorter*(k-i) + longeri;
2)第二种组合,有j块长木板,其中j>i,则短木板有k-j块,总长度为 len2 = shorter(k-j) + longerj;
两种组合下跳水板长度之差为:len2-len1 = shorter(i-j) + longer*(j-i) = (longer-shorter)*(j-i)。
由于 longer>shorter 且 j>i,因此上式的值大于 0。由此可见,任意两种不同的组合下的跳水板长度都是不一样的,而且使用的长木板越多,跳水板的长度越大。
因此创建长度为 k+1 的数组 res,对于 0≤i≤k,令 res[i]=shorter∗(k−i)+longer∗i,则 res 包含跳水板所有可能的长度,且长度为升序排序。
复杂度分析:
时间复杂度:O(k),其中 k 是木板数量。短木板和长木板一共使用 k 块,一共有 k+1 种组合,对于每种组合都要计算跳水板的长度。
空间复杂度:O(1)。除了返回值以外,额外使用的空间复杂度为常数。
代码
class Solution {
public int[] divingBoard(int shorter, int longer, int k) {
if(k==0){
return new int[0];
}
if(shorter==longer){
return new int[]{shorter*k};
}
int[] res = new int[k+1];
for(int i=0; i<=k; i++){
res[i] = i*longer + (k-i)*shorter;
}
return res;
}
}