From f9cadf34352da6388c3dea7a343bcf6d18bfd5ac Mon Sep 17 00:00:00 2001
From: Nikita Tokarchuk <nikita@tokarch.uk>
Date: Thu, 10 Feb 2022 14:24:35 +0100
Subject: [PATCH] Solution

---
 CMakeLists.txt |   6 ++
 main.cpp       | 177 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 183 insertions(+)
 create mode 100644 CMakeLists.txt
 create mode 100644 main.cpp

diff --git a/CMakeLists.txt b/CMakeLists.txt
new file mode 100644
index 0000000..f1ba86a
--- /dev/null
+++ b/CMakeLists.txt
@@ -0,0 +1,6 @@
+cmake_minimum_required(VERSION 3.21)
+project(balanced_brackets)
+
+set(CMAKE_CXX_STANDARD 14)
+
+add_executable(balanced_brackets main.cpp)
diff --git a/main.cpp b/main.cpp
new file mode 100644
index 0000000..12dc516
--- /dev/null
+++ b/main.cpp
@@ -0,0 +1,177 @@
+#include <iostream>
+#include <fstream>
+#include <vector>
+#include <array>
+#include <algorithm>
+
+template<typename T>
+class Stack {
+private:
+    size_t _len;
+    std::vector<T> _data;
+public:
+    explicit Stack(size_t alloc_size = 128);
+
+    size_t push(T elem);
+
+    T pop();
+
+    size_t len();
+};
+
+template<typename T>
+size_t Stack<T>::push(T elem) {
+
+    if (this->_data.size() > this->_len) {
+        this->_data[this->_len] = elem;
+        this->_len++;
+
+        return this->_len;
+    }
+
+    this->_data.push_back(elem);
+    this->_len = this->_data.size();
+
+    return this->_len;
+};
+
+template<typename T>
+Stack<T>::Stack(size_t alloc_size) : _len(0), _data(alloc_size) {};
+
+template<typename T>
+T Stack<T>::pop() {
+
+    if (this->_len == 0) {
+        std::abort();
+    }
+
+    this->_len--;
+
+    return this->_data[this->_len];
+}
+
+template<typename T>
+size_t Stack<T>::len() {
+    return this->_len;
+}
+
+enum class BracketType {
+    CURLY = 0,
+    SQUARE,
+    ROUND,
+    MAX,
+};
+
+enum class BracketDirection {
+    OPENING = 0,
+    CLOSING,
+    MAX,
+};
+
+typedef std::pair<BracketType, BracketDirection> Bracket_t;
+
+const std::string YES = "YES";
+const std::string NO = "NO";
+
+std::string ltrim(const std::string &);
+
+std::string rtrim(const std::string &);
+
+Bracket_t toType(char v) {
+    switch (v) {
+        case '[':
+            return {BracketType::ROUND, BracketDirection::OPENING};
+        case '{':
+            return {BracketType::CURLY, BracketDirection::OPENING};
+        case '(':
+            return {BracketType::SQUARE, BracketDirection::OPENING};
+        case ']':
+            return {BracketType::ROUND, BracketDirection::CLOSING};
+        case '}':
+            return {BracketType::CURLY, BracketDirection::CLOSING};
+        case ')':
+            return {BracketType::SQUARE, BracketDirection::CLOSING};
+        default:
+            return {BracketType::MAX, BracketDirection::MAX};
+    }
+}
+
+/*
+ * Complete the 'isBalanced' function below.
+ *
+ * The function is expected to return a std::STRING.
+ * The function accepts std::STRING s as parameter.
+ */
+
+std::string isBalanced(std::string s) {
+
+    if (s.empty()) {
+        return YES;
+    }
+
+    Stack<BracketType> brackets;
+
+    for (char i: s) {
+        Bracket_t b = toType(i);
+
+        if (b.second == BracketDirection::OPENING) {
+            brackets.push(b.first);
+            continue;
+        }
+
+        if (brackets.len() > 0) {
+            auto ctx = brackets.pop();
+            if (ctx == b.first) {
+                continue;
+            }
+        }
+
+        return NO;
+    }
+
+    return brackets.len() > 0 ? NO : YES;
+}
+
+int main() {
+    std::ofstream fout(getenv("OUTPUT_PATH"));
+
+    std::string t_temp;
+    getline(std::cin, t_temp);
+
+    int t = stoi(ltrim(rtrim(t_temp)));
+
+    for (int t_itr = 0; t_itr < t; t_itr++) {
+        std::string s;
+        getline(std::cin, s);
+
+        std::string result = isBalanced(s);
+
+        fout << result << "\n";
+    }
+
+    fout.close();
+
+    return 0;
+}
+
+std::string ltrim(const std::string &str) {
+    std::string s(str);
+
+    s.erase(
+            s.begin(),
+            find_if(s.begin(), s.end(), not1(std::ptr_fun<int, int>(isspace)))
+    );
+
+    return s;
+}
+
+std::string rtrim(const std::string &str) {
+    std::string s(str);
+
+    s.erase(
+            find_if(s.rbegin(), s.rend(), not1(std::ptr_fun<int, int>(isspace))).base(),
+            s.end()
+    );
+
+    return s;
+}