Advertisement
pratiyush7

Untitled

Feb 26th, 2022
738
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.15 KB | None | 0 0
  1. class Node {
  2.  
  3. public:
  4.  
  5.    virtual int Compute() = 0;
  6.  
  7.    inline void SetLeft(Node* left) {
  8.  
  9.       left_ = left;
  10.  
  11.    }
  12.  
  13.    inline void SetRight(Node* right) {
  14.  
  15.       right_ = right;
  16.  
  17.    }
  18.  
  19. protected:
  20.  
  21.    Node* left_ = nullptr;
  22.  
  23.    Node* right_ = nullptr;
  24.  
  25. };
  26.  
  27. class SumNode : public Node {
  28.  
  29. public:
  30.  
  31.    inline int Compute() override {
  32.  
  33.       return left_->Compute() + right_->Compute();
  34.  
  35.    }
  36.  
  37. };
  38.  
  39. class SubNode : public Node {
  40.  
  41. public:
  42.  
  43.    inline int Compute() override {
  44.  
  45.       return left_->Compute() - right_->Compute();
  46.  
  47.    }
  48.  
  49. };
  50.  
  51. class MulNode : public Node {
  52.  
  53. public:
  54.  
  55.    inline int Compute() override {
  56.  
  57.       return left_->Compute() * right_->Compute();
  58.  
  59.    }
  60.  
  61. };
  62.  
  63. class NumNode : public Node {
  64.  
  65. public:
  66.  
  67.    NumNode(int num) : num_(num) {}
  68.  
  69.    inline int Compute() override {
  70.  
  71.       return num_;
  72.  
  73.    }
  74.  
  75. private:
  76.  
  77.    int num_;
  78.  
  79. };
  80.  
  81. class Solution {
  82.  
  83. public:
  84.  
  85.    vector<int> diffWaysToCompute(string expression) {
  86.  
  87.  
  88.  
  89.       vector<Node*> nodes;
  90.  
  91.  
  92.  
  93. // parse the expression
  94.  
  95.       for (int i = 0; i < expression.size(); ++i) {
  96.  
  97.  
  98.  
  99.          switch (expression[i]) {
  100.  
  101.          case '+':
  102.  
  103.             nodes.emplace_back(new SumNode());
  104.  
  105.             break;
  106.  
  107.          case '-':
  108.  
  109.             nodes.emplace_back(new SubNode());
  110.  
  111.             break;
  112.  
  113.          case '*':
  114.  
  115.             nodes.emplace_back(new MulNode());
  116.  
  117.             break;
  118.  
  119.          default:
  120.  
  121.             int num = 0;
  122.  
  123.             while (expression[i] >= '0' && expression[i] <= '9') {
  124.  
  125.                num = num * 10 + expression[i] - '0';
  126.  
  127.                ++i;
  128.  
  129.             }
  130.  
  131.             --i;
  132.  
  133.             nodes.emplace_back(new NumNode(num));
  134.  
  135.             break;
  136.  
  137.          }
  138.  
  139.       }
  140.  
  141.  
  142.  
  143.       vector<Node*> trees = GenerateAllTrees(nodes, 0, nodes.size() - 1);
  144.  
  145.  
  146.  
  147.       for (Node* node : nodes) {
  148.  
  149.          delete node;
  150.  
  151.       }
  152.  
  153.  
  154.  
  155.       vector<int> results;
  156.  
  157.       for (Node* tree : trees) {
  158.  
  159.          results.emplace_back(tree->Compute());
  160.  
  161.          delete tree;
  162.  
  163.       }
  164.  
  165.       return results;
  166.  
  167.  
  168.  
  169.    }
  170.  
  171.  
  172.  
  173.    vector<Node*> GenerateAllTrees(const vector<Node*> nodes, int beg, int end) {
  174.  
  175.  
  176.  
  177.       if (beg == end) {
  178.  
  179.          return {new NumNode(nodes[beg]->Compute())};
  180.  
  181.       }
  182.  
  183.  
  184.  
  185.       vector<Node*> results;
  186.  
  187.  
  188.  
  189.       for (int i = beg; i <= end; i++) {
  190.  
  191.  
  192.  
  193. // assuming the expression is well formed, operators will be in odd positions
  194.  
  195.          if ((i & 1) == 0) continue;
  196.  
  197.  
  198.  
  199.          vector<Node*> left_trees = GenerateAllTrees(nodes, beg, i - 1);
  200.  
  201.          vector<Node*> right_trees = GenerateAllTrees(nodes, i + 1, end);
  202.  
  203.  
  204.  
  205.          for (Node* left : left_trees) {
  206.  
  207.             for (Node* right : right_trees) {
  208.  
  209.                nodes[i]->SetLeft(left);
  210.  
  211.                nodes[i]->SetRight(right);
  212.  
  213.                Node* result = new NumNode(nodes[i]->Compute());
  214.  
  215.                results.emplace_back(result);
  216.  
  217.             }
  218.  
  219.          }
  220.  
  221.          for (Node* t : left_trees) {
  222.  
  223.             delete t;
  224.  
  225.          }
  226.  
  227.          for (Node* t : right_trees) {
  228.  
  229.             delete t;
  230.  
  231.  
  232.  
  233.          }
  234.  
  235.  
  236.  
  237.          return results;
  238.  
  239.       }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement