给你一个数组 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]
题解 :
/**
* @param {number[]} prices
* @return {number[]}
*/
var finalPrices = function (prices) {
// 栈内存储下标
let stack = [];
for (let i = 0; i < prices.length; i++) {
// 当栈内不为空,判断prices当前值和前一位值得大小
// 当前一位大于等于prices当前值时,prices前一位赋值
// 当前一位小于prices当前值时,当前下标入栈
// 例如:prices = [8,4,6,2,3]
// 第一轮 i = 0 prices = [8, 4, 6, 2, 3] stack = [0]
// 第二轮 i = 1 prices = [4, 4, 6, 2, 3] stack = [1]
// 第三轮 i = 2 prices = [4, 4, 6, 2, 3] stack = [1, 2]
// 第四轮 i = 3 prices = [4, 2, 4, 2, 3] stack = [3]
// 第五轮 i = 4 prices = [4, 2, 4, 2, 3] stack = [3, 4]
while (stack.length && prices[stack[stack.length - 1]] >= prices[i]) {
prices[stack[stack.length - 1]] = prices[stack[stack.length - 1]] - prices[i];
stack.pop();
}
stack.push(i);
}
return prices;
};
来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。