method1
直接遍历
vector<int> finalPrices(vector<int>& prices) {
int n=prices.size();
for(int i=0; i<n-1; i++) {
for(int j=i+1; j<n; j++) {
if(prices[j]<=prices[i]) {
prices[i] -= prices[j];
break;
}
}
}
return prices;
}
method2
单调栈:由于栈是先进后出,当倒序遍历数组时,栈从上而下就是顺序排列,所以while中会查找prices[i]右侧第一个小于s的元素(并且不会pop,就不会影响其他的元素),如果while遍历后st为空证明没找到,不然st.top()就是右侧最小的元素,然后进行更新即可。
vector<int> finalPrices(vector<int>& prices) {
int n = prices.size();
int tmp;
stack<int> s;
for(int i=n-1; i>=0; i--) {
while(!s.empty() && prices[i]<s.top()) {
s.pop();
}
tmp = prices[i];
prices[i] = s.empty()?prices[i]:prices[i]-s.top();
s.emplace(tmp);
}
return prices;
}