leetcode【每日一题】面试题 16.11. 跳水板 Java

题干

你正在使用一堆木板建造跳水板。有两种类型的木板，其中长度较短的木板长度为shorter，长度较长的木板长度为longer。你必须正好使用k块木板。编写一个方法，生成跳水板所有可能的长度。

返回的长度需要从小到大排列。

示例：

输入：
shorter = 1
longer = 2
k = 3
输出： {3,4,5,6}
提示：

0 < shorter <= longer
0 <= k <= 100000

想法

首先需要搞清楚
这个组合是不会重复的，因为
考虑以下两种不同的组合：第一种组合，有 ii 块长木板，则跳水板的长度是 shorter*(k-i)+longerishorter∗(k−i)+longer∗i；第二种组合，有 jj 块长木板，则跳水板的长度是 shorter(k-j)+longer*jshorter∗(k−j)+longer∗j。其中 0 \leq i<j \leq k0≤i<j≤k。则两种不同的组合下的跳水板长度之差为：

(shorter*(k-i)+longer*i)-(shorter*(k-j)+longer*j)=(longer-shorter)*(i-j)

显然不为0

那么就遍历一次就完事儿
简单题 看代码

Java代码

package daily;

public class DivingBoard {
    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] = shorter * (k - i) + longer * i;
        }

        return res;
    }


    public static void main(String[] args) {
        DivingBoard divingBoard = new DivingBoard();
        int shorter = 1,
                longer = 2,
                k = 3;
        int[] res = divingBoard.divingBoard(shorter, longer, k);
        for (int i : res
        ) {
            System.out.print(i);
            System.out.print(',');

        }
    }

}

我的leetcode代码都已经上传到我的githttps://github.com/ragezor/leetcode