面试题-16.11

面试题-16.11 跳水板

题目

结果

代码

class Solution {
    public int[] divingBoard(int shorter, int longer, int k) {
        // 特殊情况
        if (k == 0) {
            return new int[]{};
        }
        // 特殊情况，只要shorter！=longer，数组就不会有重复元素
        if (shorter == longer) {
            return new int[]{shorter * k};
        }
        int[] ans = new int[k + 1];
        for (int i = 0; i <= k; i++) {
            ans[i] = (k - i) * shorter + i * longer;
        }
        return ans;
    }
}