题目 跳水板
你正在使用一堆木板建造跳水板。有两种类型的木板,其中长度较短的木板长度为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
思路
前一阵子难题见多了,这个题目刚看的时候还在想是不是动态规划什么的。后来仔细一想根本不需要这么麻烦。很明显,最小值就是shorter × k,最大值则是longer × k。介于中间的就是shorter × (k - 1) + longer × 1,shorter × (k - 2) + longer × 2 …………shorter × 1 + longer × (k - 1)。
那么代码实现也就是水到渠成的难度了:
class Solution {
public:
vector<int> divingBoard(int shorter, int longer, int k) {
vector<int> answer;
if (k == 0) {
return answer;
}
answer.push_back(shorter * k);
if (shorter == longer) {
return answer;
}
int minus = longer - shorter;
int min_length = shorter * k;
for (int i = 1; i <= k; i++) {
answer.push_back(min_length + minus * i);
}
return answer;
}
};
其中k居然可以为0,注意判断下边界条件即可。