AVLTreeElement.h

Go to the documentation of this file.

#ifndef AVL_ELEMENT_H
#define AVL_ELEMENT_H

#include "BSTElement.h" //string, sstream, using std

namespace bridges {
    namespace datastructure {
        template <typename K, typename E>
        class AVLTreeElement : public BSTElement<K, E> {
            private:
                // The height and balance factor values
                int height = 0, balFactor = 0;
            public:
                AVLTreeElement(const K& k, const E& val = E(), const string& lab = string())
                    : BSTElement<K, E>(k, nullptr, nullptr, val, lab) {}

                virtual const string getDStype() const override {
                    return "AVLTree";
                }
                int getHeight() const {
                    return height;
                }
                void setHeight(const int& h) {
                    height = h;
                }
                int getBalanceFactor() const {
                    return balFactor;
                }
                void setBalanceFactor(const int& bf) {
                    balFactor = bf;
                }
                virtual AVLTreeElement* getLeft() override {
                    return static_cast<AVLTreeElement*>(BSTElement<K, E>::getLeft());
                }
                virtual const AVLTreeElement* getLeft() const override {
                    return static_cast<const AVLTreeElement*>(BSTElement<K, E>::getLeft());
                }
                void setLeft(AVLTreeElement* l) {
                    BSTElement<K, E>::setLeft(l);
                }
                virtual AVLTreeElement* getRight() override {
                    return static_cast<AVLTreeElement*>(BSTElement<K, E>::getRight());
                }
                virtual const AVLTreeElement* getRight() const override {
                    return static_cast<const AVLTreeElement*>(BSTElement<K, E>::getRight());
                }
                void setRight(AVLTreeElement* r) {
                    BSTElement<K, E>::setRight(r);
                }
            private:
                virtual const string getElementRepresentation() const override final {
                    string json = BSTElement<K, E>::getElementRepresentation();

                    return json.insert (
                            json.size() - 1,
                            COMMA + QUOTE + "height" + QUOTE + COLON + to_string(height) +
                            COMMA + QUOTE + "balance_factor" + QUOTE + COLON +
                            to_string(balFactor)
                        );
                }
        }; //end of BSTElement class

    }
}//end of bridges namespace
#endif