css343_Encoding/TreeNode.h at main · marlinism/css343_Encoding · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#ifndef TREENODE_H
#define TREENODE_H

#include <memory>
#include <iostream>
#include <vector>
#include <utility>
#include <string>
#include <iterator>
#include <queue>
#include <algorithm>
#include <bitset>

using namespace std;

class TreeNode {
public:

    char letter;
    double freq;
    shared_ptr<TreeNode> left;
    shared_ptr<TreeNode> right;

  TreeNode();

  // Create T
  TreeNode(char l, double f);

  // Generate tree
  TreeNode generateTree(TreeNode newNode1, TreeNode newNode2);

  char reference(TreeNode newNode1, TreeNode newNode2);

  bool isLeaf(shared_ptr<TreeNode> root);
};

class Tree {
public:

private:
  shared_ptr<TreeNode> root;
};


#endif // TREENODE.H