[LeetCode]（面试题 16.11）跳水板

题目

你正在使用一堆木板建造跳水板。有两种类型的木板，其中长度较短的木板长度为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;
    }
}