#pragma once
// 定义AVL树节点结构
template<class K, class V>
struct AVLTreeNode
{// 节点默认构造函数AVLTreeNode(): _left(nullptr), _right(nullptr), _parent(nullptr), _by(0){}AVLTreeNode<K, V>* _left;       // 左子节点指针AVLTreeNode<K, V>* _right;      // 右子节点指针AVLTreeNode<K, V>* _parent;     // 父节点指针int _by;                        // 平衡因子pair<K, V> _kv;                 // 节点存储的关键字和值
};// 定义AVL树类
template<class K, class V>
class AVLTree
{typedef AVLTreeNode<K, V> Node;
public:// 插入节点方法// @param data: 需要插入的数据对// @return: 插入成功返回truebool insert(const pair<K, V>& data){// 如果树为空，直接创建并返回新节点if (_root == nullptr){_root = new Node(data);return true;}Node* parent = nullptr;       // 插入节点的父节点指针Node* cur = _root;            // 当前遍历节点指针// 遍历查找插入位置while (cur){if (cur->_kv.first > data.first){parent = cur;cur = cur->_left;}else if (cur->_kv.first <= data.first){parent = cur;cur = cur->_right;}}// 创建新节点并连接到父节点cur = new Node(data);if (parent->_kv.first > data.first){parent->_left = cur;cur->_parent = parent;}else{parent->_right = cur;cur->_parent = parent;}while (parent){if(cur == parent->_right){parent->_by++;}else{parent->_by--;}if(parent->_by == 0) break;parent = parent->_parent;   // 父节点平衡因子为0，则平衡}return true;}private:Node* _root;                     // 树的根节点指针
};