算法训练营 第十一天
栈与队列 day02~
上期回顾
前两天周末玩疯了,少打了两天卡,栈与队列第一天的内容也有点忘记了,所以先回顾了一下上一次的内容。
deque——双端数组,可以头插,不是连续存储空间
queue——队列,先进先出
stack——栈,后进先出
20. 有效的括号
这道题用栈来实现,如果是左括号就存入栈,如果是右括号就和栈顶元素匹配,匹配成功就出栈,不成功就返回false。有几种特殊情况
1、栈已经空了,但输入还没结束,即,多了几个右括号的情况。
2、输入数量为奇数时,直接false。
class Solution {
public:
bool isValid(string s) {
stack<char> stk;
if (s.size() % 2 != 0)
return false;
for (auto a : s) {
if (a == '[')
stk.push(']');
else if (a == '(')
stk.push(')');
else if (a == '{')
stk.push('}');
else {
if (stk.empty())
return false;
if (a == stk.top())
stk.pop();
else
return false;
}
}
if (!stk.empty())
return false;
return true;
}
};
参考答案
class Solution {
public:
bool isValid(string s) {
if (s.size() % 2 != 0) return false; // 如果s的长度为奇数,一定不符合要求
stack<char> st;
for (int i = 0; i < s.size(); i++) {
if (s[i] == '(') st.push(')');
else if (s[i] == '{') st.push('}');
else if (s[i] == '[') st.push(']');
// 第三种情况:遍历字符串匹配的过程中,栈已经为空了,没有匹配的字符了,说明右括号没有找到对应的左括号 return false
// 第二种情况:遍历字符串匹配的过程中,发现栈里没有我们要匹配的字符。所以return false
else if (st.empty() || st.top() != s[i]) return false;
else st.pop(); // st.top() 与 s[i]相等,栈弹出元素
}
// 第一种情况:此时我们已经遍历完了字符串,但是栈不为空,说明有相应的左括号没有右括号来匹配,所以return false,否则就return true
return st.empty();
}
};
参考答案和我的答案逻辑是一样的, 但是代码会更简洁,有值得学习的地方。
1047. 删除字符串中的所有相邻重复项
这道题,看着就是用栈的题,用栈实现了一下,但是这道题用string就行,用栈实现了之后,最后还得把栈里的内容倒到string里,那不如最开始就用string;写完发现,用双指针也行,这样就不用再创建一个string容器了
栈实现
class Solution {
public:
string removeDuplicates(string s) {
string res;
stack<char> st;
for (int i = s.size() - 1; i >= 0; i--) {
if (!st.empty() && s[i] == st.top())
st.pop();
else
st.push(s[i]);
}
while (!st.empty()) {
res.push_back(st.top());
st.pop();
}
return res;
}
};
string实现
class Solution {
public:
string removeDuplicates(string s) {
string res;
for (int i = 0; i <s.size(); i++) {
if (!res.empty() && s[i] == res.back())
res.resize(res.size() - 1);
else
res.push_back(s[i]);
}
return res;
}
};
双指针实现
class Solution {
public:
string removeDuplicates(string s) {
int slow = 0;
for (int i = 1; i <s.size(); i++) {
if (slow < 0) {
slow++;
s[slow] = s[i];
}
else if (s[slow] == s[i]) {
slow--;
}
else {
s[++slow] = s[i];
}
}
s.resize(slow + 1);
return s;
}
};
然后去看了答案,发现和我的想法不谋而合,代码随想录中是用string模拟栈实现的,还没有用到双指针。
参考答案
class Solution {
public:
string removeDuplicates(string S) {
string result;
for(char s : S) {
if(result.empty() || result.back() != s) {
result.push_back(s);
}
else {
result.pop_back();
}
}
return result;
}
};
学到的东西:string里的pop_back()函数可以删除字符串的尾端元素,我用的resize(),其实pop_back()会更好一些。
150. 逆波兰表达式求值
这道题,也是典型的,一看就知道得用stack来做。
我的答案
class Solution {
public:
int evalRPN(vector<string>& tokens) {
int res = 0;
stack<int> sk;
for (auto a : tokens) {
if (a == "+") {
int b = sk.top();sk.pop();
int c = sk.top();sk.pop();
sk.push(b + c);
}
else if (a == "-") {
int b = sk.top();sk.pop();
int c = sk.top();sk.pop();
sk.push(c - b);
}
else if (a == "*") {
int b = sk.top();sk.pop();
int c = sk.top();sk.pop();
sk.push(c * b);
}
else if (a == "/") {
int b = sk.top();sk.pop();
int c = sk.top();sk.pop();
sk.push(c / b);
}
else {
sk.push(stoi(a));
}
}
return sk.top();
}
};
参考答案
class Solution {
public:
int evalRPN(vector<string>& tokens) {
// 力扣修改了后台测试数据,需要用longlong
stack<long long> st;
for (int i = 0; i < tokens.size(); i++) {
if (tokens[i] == "+" || tokens[i] == "-" || tokens[i] == "*" || tokens[i] == "/") {
long long num1 = st.top();
st.pop();
long long num2 = st.top();
st.pop();
if (tokens[i] == "+") st.push(num2 + num1);
if (tokens[i] == "-") st.push(num2 - num1);
if (tokens[i] == "*") st.push(num2 * num1);
if (tokens[i] == "/") st.push(num2 / num1);
} else {
st.push(stoll(tokens[i]));
}
}
int result = st.top();
st.pop(); // 把栈里最后一个元素弹出(其实不弹出也没事)
return result;
}
};
这道题没什么好说的,知道是用栈来写之后,逻辑还是很明了的。参考答案的代码比我的要简洁一些,但复杂度是一样的。