testlab2/MyTree.java at master · Alex-aka-ion/testlab2 · 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
package com.company;

public class MyTree {
    private MyNode root;

    MyTree (int[] inputList) {
        boolean first = true;
        for (int value : inputList) {
            if (first) {
                root = new MyNode(value);
                first = false;
            } else {
                root.addBranch(value);
            }
        }
    }

    boolean contains(int v) {
        MyNode tmpNode = this.root;

        while (tmpNode != null) {
            if (tmpNode.getValue() == v) {
                return true;
            } else {
                tmpNode = tmpNode.chooseNextNode(v);
            }
        }
        return false;
    }

    int count(int v) {
        int result = 0;
        MyNode tmpNode = this.root;

        while (tmpNode != null) {
            if (tmpNode.getValue() == v) {
                result++;
            }
            tmpNode = tmpNode.chooseNextNode(v);
        }
        return result;
    }
}