Leetcode面试题16.11:跳水板

面试题16.11:跳水板

    • 题目
    • 思路一——暴力

题目

你正在使用一堆木板建造跳水板。有两种类型的木板,其中长度较短的木板长度为shorter,长度较长的木板长度为longer。你必须正好使用k块木板。编写一个方法,生成跳水板所有可能的长度。
返回的长度需要从小到大排列。
示例:

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

提示:

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

思路一——暴力

由于需要K个木板,首先需要排除特殊情况:

  • 当 K = 0,返回空;
  • 当shorter == longer 为真时,返回长度为1的数组,且值为shorter * k;

然后,由于有shorter 和longer两个值,因此每一个值范围为0~k,则返回数组长度为k+1,一次遍历即可,每次的值为 shorter * i + (k - i) * longer;
最后,题目要求返回结果需从小到大排序,则调用Arrays.sort(ret)即可。

class Solution {
    public int[] divingBoard(int shorter, int longer, int k) {
        //step1:判断特殊情况
        if(k == 0) return new int[]{};
        if(shorter == longer) return new int[]{ shorter * k};
        //Step2:定义结果变量
        int[] ret = new int[ k + 1];
        //Step3:一次遍历
        for(int i = 0; i <= k; i++){
            ret[i] = i * shorter + (k - i) * longer;
        }
        //Step4:排序
        Arrays.sort(ret);
        //Step5:返回结果
        return ret;
    }
    
}

Leetcode面试题16.11:跳水板_第1张图片
如何降低执行时间:
每次都需要执行i * shorter + (k - 1) * longer,四次运算,通过观察发现,i递增时,都只增加了(longer - shorter),可以提前定义一个变量 diff = longer - shorter ,每次当I 递增时,都用ret[I - 1] + diff,只需要两次加法操作,整体运行时间提高了3ms.

		int diff = longer - shorter;
        ret[0] = k * shorter;
        //Step3:一次遍历
        for(int i = 1; i <= k; i++){
            ret[i] = ret[i - 1] + diff;
        }

Leetcode面试题16.11:跳水板_第2张图片
时间复杂度:O(N)

你可能感兴趣的:(leedcode刷题,数组)