150. 逆波兰表达式求值
后缀表达式求值,借助栈,遇到符号就从栈顶弹出两个元素进行对应操作。
需要注意的是,从栈中先弹出的元素是操作符右侧的元素, 后弹出的是操作符左侧的元素。
时间复杂度:
O
(
n
)
O(n)
O(n)
空间复杂度:
O
(
n
)
O(n)
O(n)
// c++
class Solution {
public:
int evalRPN(vector<string>& tokens) {
stack<int> st;
for(int i=0; i<tokens.size(); i++){
if(!st.empty() && (tokens[i]=="+"||tokens[i]=="-"||tokens[i]=="*"||tokens[i]=="/")) {
int res2 = st.top();
st.pop();
int res1 = st.top();
st.pop();
if(tokens[i]=="+")
res1 += res2;
else if(tokens[i]=="-")
res1 -= res2;
else if(tokens[i]=="*")
res1 *= res2;
else if(tokens[i]=="/")
res1 /= res2;
st.push(res1);
}else{
st.push(stoi(tokens[i]));
}
}
return st.top();
}
};
*239. 滑动窗口最大值
思路来源
使用单调队列,队首始终存放最大值。
时间复杂度:
O
(
n
)
O(n)
O(n)
空间复杂度:
O
(
k
)
O(k)
O(k)
// c++
class Solution {
private:
class MyQue{
public:
deque<int> que;
void pop(int val){
if(!que.empty() && val == que.front()){
que.pop_front();
}
}
void push(int val){
while(!que.empty() && que.back()<val){
que.pop_back();
}
que.push_back(val);
}
int front(){
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQue que;
vector<int> result;
int i=0;
while(i<k){
que.push(nums[i]);
i++;
}
result.emplace_back(que.front());
for(;i<nums.size(); i++){
que.pop(nums[i-k]);
que.push(nums[i]);
result.emplace_back(que.front());
}
return result;
}
};
*347. 前 K 个高频元素
思路来源
先用map记录每个元素的频次,再使用小顶堆来选出前K个高频的元素。
时间复杂度:
O
(
n
l
o
g
k
)
O(nlogk)
O(nlogk)
空间复杂度:
O
(
n
)
O(n)
O(n)
// c++
class Solution {
private:
// 小顶堆
class cmp {
public:
bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
return lhs.second > rhs.second;
}
};
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> hash;
for(int i=0; i<nums.size(); i++) hash[nums[i]]++;
priority_queue<pair<int, int>, vector<pair<int,int>>, cmp> pri_que;
for(auto it=hash.begin(); it!=hash.end(); it++){
pri_que.push(*it);
if(pri_que.size() > k) pri_que.pop();
}
vector<int> result;
while(!pri_que.empty()){
result.emplace_back(pri_que.top().first);
pri_que.pop();
}
return result;
}
};