LeetCode_面试题_16.11_跳水板

首先看题目：

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