-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrpn150.cpp
63 lines (56 loc) · 1.38 KB
/
rpn150.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#include <bits/stdc++.h>
using namespace std;
template <typename... T>
void pln(T... args)
{
(cout << ... << args) << "\n";
}
void test(bool cond)
{
if (cond) pln("Passed!!");
else pln("Failed!!");
}
//----------------------------------------------------------------------------------
// #pragma GCC optimize("O3")
namespace {
static const bool __booster = [] {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
return true;
}();
} // namespace
class Solution {
int op(stack<int> &s, const string &o)
{
int y = s.top();
s.pop();
int x = s.top();
s.pop();
if (o == "+") return x + y;
else if (o == "-") return x - y;
else if (o == "*") return x * y;
else if (o == "/") return x / y;
return 0;
}
public:
int evalRPN(vector<string> &tokens)
{
if (tokens.size() == 1) return atoi(tokens.front().c_str());
stack<int> tStack;
for (size_t i = 0; i < tokens.size() - 1; i++) {
auto s = tokens[i];
if (s != "+" && s != "-" && s != "*" && s != "/") tStack.push(atoi(s.c_str()));
else tStack.push(op(tStack, s));
}
return op(tStack, tokens.back());
}
};
//----------------------------------------------------------------------------------
int main([[maybe_unused]] int argc, [[maybe_unused]] char **argv)
{
Solution s;
vector<string> v1 = {"2", "1", "+", "3", "*"};
s.evalRPN(v1);
return 0;
}