[心得] [演算法心得] ZeroJudge f640.函數運算式求值

1. #include<bits/stdc++.h>
   
2. #define ll long long
   
3. #define INF 0x3f3f3f3f
   
4. #define MOD 1000000007
   
5. #define io iOS::sync_with_stdio(0);cin.tie(0);cout.tie(0);
   
6. using namespace std;
   
7. const int MXN = 200005;
   
8. 
   
9. //int ar[MXN];
   
10. 
    
11. int f(int x){
    
12.   return 2 * x - 3;
    
13. }
    
14. 
    
15. int g(int x, int y){
    
16.   return 2 * x + y - 7;
    
17. }
    
18. 
    
19. int h(int x, int y, int z){
    
20.   return 3 * x - 2 * y + z;
    
21. }
    
22. 
    
23. int toInt(string s){
    
24.   int cur = 0, op = 1;
    
25.   for(char i : s){
    
26.     if(i == '-'){
    
27.       op = -1;
    
28.       continue;
    
29.     }
    
30.     cur = cur * 10 + i - '0';
    
31.   }
    
32.   return cur * op;
    
33. }
    
34. 
    
35. void sol(){
    
36.   ll ans = 0, n = 0, m = 0, k = 0;
    
37.   
    
38.   string s;
    
39.   vector<string> vec;
    
40.   stack<int> st;
    
41.   while(cin >> s){
    
42.     vec.emplace_back(s);
    
43.   }
    
44. 
    
45.   for(string i : vector<string>(vec.rbegin(), vec.rend())){
    
46.     if(i == "f"){
    
47.       int a = st.top(); st.pop();
    
48.       st.push(f(a));
    
49.     }
    
50.     else if(i == "g"){
    
51.       int a = st.top(); st.pop();
    
52.       int b = st.top(); st.pop();
    
53.       st.push(g(a, b));
    
54.     }
    
55.     else if(i == "h"){
    
56.       int a = st.top(); st.pop();
    
57.       int b = st.top(); st.pop();
    
58.       int c = st.top(); st.pop();
    
59.       st.push(h(a, b, c));
    
60.     }
    
61.     else{
    
62.       st.push(toInt(i));
    
63.     }
    
64.   }
    
65.   cout << st.top() << '\n';
    
66. }
    
67. 
    
68. int main(){
    
69.   //io
    
70.   int t = 1;
    
71.   while(t--){
    
72.     sol();
    
73.   }
    
74. }

複製代碼