二进制搜索树迭代非键

时间:2014-07-30 21:24:30

标签: java printing binary-search-tree

我有点难题。我需要打印我的BST中不是键的所有值。因此,由于树没有按照这些值排序,所以我不能像过去那样通常使用BST。我只需要查看树上的每个节点,将非键值与我输入的值进行比较,并确定是否打印它。

即。学生目录,我需要打印高于2.0的所有GPA。由于树是按学生ID而非GPA排序的,如何通过每个节点并比较GPA并打印所有高于2.0的节点?

如果你需要查看我的代码,整个过程就在这里,而且它是巨大的。

public class StudentBST
{
private static Node root;

static class Node
{
    public int studentID;
    public String lastName;
    public String firstName;
    public String major;
    public double gpa;
    public Node left, right;

    public int minValue()
    {
        if(left == null)
        {
            return studentID;
        }
        else
        {
            return left.minValue();
        }
    }

    public boolean remove(int i, Node node)
    {
        if(i < this.studentID)
        {
            if(left != null)
            {
                return left.remove(i, this);
            }
            else
            {
                return false;
            }
        }
        else if(i > this.studentID)
        {
            if(right != null)
            {
                return right.remove(i, this);
            }
            else
            {
                return false;
            }
        }
        else
        {
            if(left != null && right != null)
            {
                this.studentID = right.minValue();
                right.remove(this.studentID, this);
            }
            else if(node.left == this)
            {
                node.left = (left != null) ? left : right;
            }
            else if(node.right == this)
            {
                node.right = (left != null) ? left : right;
            }
            return true;
        }
    }

    public Node(int i, String l, String f, String m, double g)
    {
    studentID = i;
    lastName = l;
    firstName = f;
    major = m;
    gpa = g;
    left = null;
    right = null;
    }
}
public StudentBST()
{
    root = null;
}
private static void insert(int i, String l, String f, String m, double g)
{
    root = insert(root, i, l, f, m , g);
}
private static Node insert(Node node, int i, String l, String f, String m, double g)
{
    if(node == null)
    {
        node = new Node(i, l, f, m, g);
    }
    else
    {
        if(i <= node.studentID)
        {
            node.left = insert(node.left, i, l, f, m, g);
        }
        else
        {
            node.right = insert(node.right, i, l, f, m, g);
        }
    }
    return(node);
}
public static void printBST()
{
    printBST(root);
    System.out.println();
}
private static void printBST(Node node)
{
    if(node == null)
    {
        return;
    }
    printBST(node.left);
    System.out.println(node.studentID + ", " + node.lastName + ", " + node.firstName
    + ", " + node.major + ", " + node.gpa);
    printBST(node.right);
}
public static boolean remove(int i)
{
    if(root == null)
    {
        return false;
    }
    else
    {
        if(root.studentID == i)
        {
            Node auxRoot = new Node(0, "", "", "", 0);
            auxRoot.left = root;
            boolean result = root.remove(i, auxRoot);
            root = auxRoot.left;
            return result;
        }
        else
        {
            return root.remove(i, null);
        }
    }
}
public static void main(String[] args) 
{
    StudentBST.insert(8, "Costanza", "George", "Napping", 1.60);
    StudentBST.insert(10, "Kramer", "Cosmo", "Chemistry", 3.04);
    StudentBST.insert(5, "Seinfeld", "Jerry", "Theater", 2.05);

    StudentBST.printBST();
    Scanner input = new Scanner(System.in);
    int option = 9;

    while(option != 0)
    {
        System.out.println("1 - Add new student 2 - Delete student 3 - Print All" + 
                " 0 - Exit");
        option = input.nextInt();

        if(option == 1)
        {
            System.out.println("Enter student ID");
            int i = input.nextInt();
            input.nextLine();

            System.out.println("Enter Last Name");
            String l = input.nextLine();

            System.out.println("Enter First Name");
            String f = input.nextLine();

            System.out.println("Enter major");
            String m = input.nextLine();

            System.out.println("Enter GPA");
            Double g = input.nextDouble();

            System.out.println("Inserted student record");
            StudentBST.insert(i, l, f, m, g);
        }
        if(option == 2)
        {
            System.out.println("Enter Student ID to delete");
            int i = input.nextInt();
            boolean b = StudentBST.remove(i);
            if(b)
            {
                System.out.println("Deletion completed");
            }
            else
            {
                System.out.println("Deletion encountered error");
            }
        }
        if(option == 3)
        {
            StudentBST.printBST();
        }
    }
}

1 个答案:

答案 0 :(得分:3)

我认为你有正确的想法:只需遍历整个树并打印出高于某个阈值的GPA。粗略的实现如下:

public void printGPAs(Node node, double gpa_cutoff) {
    if (node == null) {
        return;
    } 

    if (node.gpa >= gpa_cutoff) {
        System.out.println(node.gpa);
    }

    printGPAs(node.left);
    printGPAs(node.right);
}

如果您想按特定顺序打印出来,最简单的方法是将它们放入列表中,插入正确的位置以保持所需的顺序。