题目
给你一个数组
prices,其中prices[i]是商店里第i件商品的价格。商店里正在进行促销活动,如果你要买第
i件商品,那么你可以得到与prices[j]相等的折扣,其中j是满足j > i且prices[j] <= prices[i]的 最小下标 ,如果没有满足条件的j,你将没有任何折扣。请你返回一个数组,数组中第
i个元素是折扣后你购买商品i最终需要支付的价格。
示例1.
输入:prices = [8,4,6,2,3] 输出:[4,2,4,2,3] 解释: 商品 0 的价格为 price[0]=8 ,你将得到 prices[1]=4 的折扣,所以最终价格为 8 - 4 = 4 。 商品 1 的价格为 price[1]=4 ,你将得到 prices[3]=2 的折扣,所以最终价格为 4 - 2 = 2 。 商品 2 的价格为 price[2]=6 ,你将得到 prices[3]=2 的折扣,所以最终价格为 6 - 2 = 4 。 商品 3 和 4 都没有折扣。
示例2.
输入: prices = [1,2,3,4,5] 输出: [1,2,3,4,5] 解释: 在这个例子中,所有商品都没有折扣。
示例3.
输入: prices = [10,1,1,6] 输出: [9,0,1,6]
- 难度:中等
- 使用语言:TypeScript
- 链接: 商品折扣后的最终价格
九月份的第一天,从一道简单的题目开始😁。
解法1
这道题目有两种解法,一种是直接暴力求解,每次遍历的时候都从当前数字开始,往后一个个找,找到一个比当前数字小的数字,就减去他并跳出循环。
最差的情况就像示例二那样,单调递增的数组,这种时间复杂度为n^2。
function finalPrices(prices: number[]): number[] {
for (let i = 0; i < prices.length; i++){
for (let j = i + 1; j < prices.length; j++) {
if (prices[i] >= prices[j]) {
prices[i] -= prices[j]
break
}
}
}
return prices
};
解法2
时间复杂度为n^2还是太高了,能不能优化一下呢?
昨天的每日一题练习用到了栈来解决问题,这道题其实也可以。
这道题关键在于要找到一个合适的数字减掉,他需要满足:
1.在这个数字后面。
2.要比这个数字小。
3.要尽可能离他近。
因为要减掉的数字在后面,所以我们可以吧数据从后往前遍历,记录一个从单调递增的栈。
入栈的值必须是栈的最大值,如果栈顶的值比入栈的值要大,那么要先出栈,直到栈顶的值比入栈的值要小。
每个数字都减去栈顶的值。如果栈为空,则保持不变(减去0)
上代码:
function finalPrices(prices: number[]): number[] {
let res: number[] = []
let stack: number[] = []
for (let i = prices.length - 1; i >= 0; i--) {
while (stack.length && stack[stack.length - 1] > prices[i]) {
stack.pop()
}
res[i] = prices[i] - (stack[stack.length - 1] || 0)
stack.push(prices[i])
}
return res
};
总结
- 可以先尝试一下暴力求解,这种解法一般很简单,但是时间复杂度偏高。
- 暴力求解做出来后想想有没有办法可以降低一下时间复杂度,如果没想到可以去题解里面逛逛,说不定能找到很多好方法。
- 使用栈/队列的解法一般来说时间复杂度不会太高,没想到好方法可以试试看能不能套进去,就和做数学题一样,做多了就有思路了。