力扣每日一题:(1475. 商品折扣后的最终价格)

212 阅读3分钟

题目

给你一个数组 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]

九月份的第一天,从一道简单的题目开始😁。

解法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
};

总结

  • 可以先尝试一下暴力求解,这种解法一般很简单,但是时间复杂度偏高。
  • 暴力求解做出来后想想有没有办法可以降低一下时间复杂度,如果没想到可以去题解里面逛逛,说不定能找到很多好方法。
  • 使用栈/队列的解法一般来说时间复杂度不会太高,没想到好方法可以试试看能不能套进去,就和做数学题一样,做多了就有思路了。