Naloga 3
This commit is contained in:
		
							parent
							
								
									1f2ae5b63a
								
							
						
					
					
						commit
						97c67bea67
					
				@ -1,12 +1,13 @@
 | 
			
		||||
@SuppressWarnings("SuspiciousNameCombination")
 | 
			
		||||
public class AVLDrevo {
 | 
			
		||||
    static class Item {
 | 
			
		||||
        public int value;
 | 
			
		||||
        public int count;
 | 
			
		||||
    static class Node {
 | 
			
		||||
        int value;
 | 
			
		||||
        int count = 1;
 | 
			
		||||
        Node left, right;
 | 
			
		||||
        int height = 1;
 | 
			
		||||
 | 
			
		||||
        public Item(int value) {
 | 
			
		||||
        public Node(int value) {
 | 
			
		||||
            this.value = value;
 | 
			
		||||
            this.count = 1;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        @Override
 | 
			
		||||
@ -15,18 +16,8 @@ public class AVLDrevo {
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    static class Node {
 | 
			
		||||
        Item item;
 | 
			
		||||
        Node left, right;
 | 
			
		||||
        int height = 1;
 | 
			
		||||
 | 
			
		||||
        public Node(int value) {
 | 
			
		||||
            item = new Item(value);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    Node root;
 | 
			
		||||
    StringBuilder traversal;
 | 
			
		||||
    private StringBuilder traversal = new StringBuilder();
 | 
			
		||||
 | 
			
		||||
    private int getHeight(Node node) {
 | 
			
		||||
        if (node == null) {
 | 
			
		||||
@ -60,10 +51,10 @@ public class AVLDrevo {
 | 
			
		||||
        return y;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private Node minValueNode(Node node) {
 | 
			
		||||
    private Node maxValueNode(Node node) {
 | 
			
		||||
        Node current = node;
 | 
			
		||||
        while (current.left != null) {
 | 
			
		||||
            current = current.left;
 | 
			
		||||
        while (current.right != null) {
 | 
			
		||||
            current = current.right;
 | 
			
		||||
        }
 | 
			
		||||
        return current;
 | 
			
		||||
    }
 | 
			
		||||
@ -101,10 +92,10 @@ public class AVLDrevo {
 | 
			
		||||
            return new Node(value);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if (value == node.item.value) {
 | 
			
		||||
            node.item.count++;
 | 
			
		||||
        if (value == node.value) {
 | 
			
		||||
            node.count++;
 | 
			
		||||
            return node;
 | 
			
		||||
        } else if (value < node.item.value) {
 | 
			
		||||
        } else if (value < node.value) {
 | 
			
		||||
            node.left = vstavi(node.left, value);
 | 
			
		||||
        } else {
 | 
			
		||||
            node.right = vstavi(node.right, value);
 | 
			
		||||
@ -121,26 +112,26 @@ public class AVLDrevo {
 | 
			
		||||
        System.out.println(traversal.substring(0, traversal.length() - 1));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private void najdi(Node node, int value) {
 | 
			
		||||
    private Node najdi(Node node, int value) {
 | 
			
		||||
        if (node == null) {
 | 
			
		||||
            traversal.append("x").append(",");
 | 
			
		||||
            return;
 | 
			
		||||
            return null;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if (value == node.item.value) {
 | 
			
		||||
            traversal.append(node.item.value).append(",");
 | 
			
		||||
            return;
 | 
			
		||||
        } else if (value < node.item.value) {
 | 
			
		||||
            traversal.append(node.item.value).append(",");
 | 
			
		||||
            najdi(node.left, value);
 | 
			
		||||
        if (value == node.value) {
 | 
			
		||||
            traversal.append(node.value).append(",");
 | 
			
		||||
            return node;
 | 
			
		||||
        } else if (value < node.value) {
 | 
			
		||||
            traversal.append(node.value).append(",");
 | 
			
		||||
            return najdi(node.left, value);
 | 
			
		||||
        } else {
 | 
			
		||||
            traversal.append(node.item.value).append(",");
 | 
			
		||||
            najdi(node.right, value);
 | 
			
		||||
            traversal.append(node.value).append(",");
 | 
			
		||||
            return najdi(node.right, value);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    public void izbrisi(int kljuc) {
 | 
			
		||||
        izbrisi(root, kljuc);
 | 
			
		||||
        root = izbrisi(root, kljuc);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private Node izbrisi(Node node, int value) {
 | 
			
		||||
@ -148,28 +139,27 @@ public class AVLDrevo {
 | 
			
		||||
            return null;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if (value < node.item.value) {
 | 
			
		||||
        if (value < node.value) {
 | 
			
		||||
            node.left = izbrisi(node.left, value);
 | 
			
		||||
        } else if (value > node.item.value) {
 | 
			
		||||
        } else if (value > node.value) {
 | 
			
		||||
            node.right = izbrisi(node.right, value);
 | 
			
		||||
        } else {
 | 
			
		||||
            if (node.item.count > 1) {
 | 
			
		||||
                node.item.count--;
 | 
			
		||||
            if (node.count > 1) {
 | 
			
		||||
                node.count--;
 | 
			
		||||
                return node;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            Node tmp;
 | 
			
		||||
            if (node.left == null) {
 | 
			
		||||
                tmp = node.right;
 | 
			
		||||
                return tmp;
 | 
			
		||||
                return node.right;
 | 
			
		||||
            } else if (node.right == null) {
 | 
			
		||||
                tmp = node.left;
 | 
			
		||||
                return tmp;
 | 
			
		||||
                return node.left;
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            tmp = minValueNode(node.right);
 | 
			
		||||
            node.item = tmp.item;
 | 
			
		||||
            node.right = izbrisi(node.right, tmp.item.value);
 | 
			
		||||
            Node max = maxValueNode(node.left);
 | 
			
		||||
            node.value = max.value;
 | 
			
		||||
            node.count = max.count;
 | 
			
		||||
            max.count = 1;
 | 
			
		||||
            node.left = izbrisi(node.left, max.value);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        node.height = 1 + Math.max(getHeight(node.left), getHeight(node.right));
 | 
			
		||||
@ -191,18 +181,23 @@ public class AVLDrevo {
 | 
			
		||||
        if (node == null) {
 | 
			
		||||
            return;
 | 
			
		||||
        }
 | 
			
		||||
        traversal.append(node.item).append(",");
 | 
			
		||||
        traversal.append(node).append(",");
 | 
			
		||||
        premiPregled(node.left);
 | 
			
		||||
        premiPregled(node.right);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    public void najnizjiSkupniPredhodnik(int a, int b) {
 | 
			
		||||
        if (najdi(root, a) == null || najdi(root, b) == null) {
 | 
			
		||||
            System.out.println("x");
 | 
			
		||||
            return;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        Node node = lca(root, a, b);
 | 
			
		||||
        if (node == null) {
 | 
			
		||||
            System.out.println("x");
 | 
			
		||||
            return;
 | 
			
		||||
        }
 | 
			
		||||
        System.out.println(node.item.value);
 | 
			
		||||
        System.out.println(node.value);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private Node lca(Node node, int a, int b) {
 | 
			
		||||
@ -210,9 +205,9 @@ public class AVLDrevo {
 | 
			
		||||
            return null;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if (a < node.item.value && b < node.item.value) {
 | 
			
		||||
        if (a < node.value && b < node.value) {
 | 
			
		||||
            return lca(node.left, a, b);
 | 
			
		||||
        } else if (a > node.item.value && b > node.item.value) {
 | 
			
		||||
        } else if (a > node.value && b > node.value) {
 | 
			
		||||
            return lca(node.right, a, b);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
@ -230,8 +225,8 @@ public class AVLDrevo {
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        int sum = 0;
 | 
			
		||||
        if (node.item.value >= lower && node.item.value <= upper) {
 | 
			
		||||
            sum += node.item.value * node.item.count;
 | 
			
		||||
        if (node.value >= lower && node.value <= upper) {
 | 
			
		||||
            sum += node.value * node.count;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        sum += sumBounds(node.left, lower, upper);
 | 
			
		||||
@ -255,11 +250,23 @@ public class AVLDrevo {
 | 
			
		||||
        if (k == null) {
 | 
			
		||||
            return null;
 | 
			
		||||
        }
 | 
			
		||||
        k -= node.item.count;
 | 
			
		||||
        k -= node.count;
 | 
			
		||||
        if (k <= 0) {
 | 
			
		||||
            System.out.println(node.item.value);
 | 
			
		||||
            System.out.println(node.value);
 | 
			
		||||
            return null;
 | 
			
		||||
        }
 | 
			
		||||
        return kSmallest(node.right, k);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    static void beautyPrint(AVLDrevo.Node node, int l) {
 | 
			
		||||
        if (node == null)
 | 
			
		||||
            return;
 | 
			
		||||
 | 
			
		||||
        for (int i = 0; i < l; i++) {
 | 
			
		||||
            System.out.print("\t");
 | 
			
		||||
        }
 | 
			
		||||
        System.out.println(node);
 | 
			
		||||
        beautyPrint(node.left, l + 1);
 | 
			
		||||
        beautyPrint(node.right, l + 1);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@ -1,16 +1,28 @@
 | 
			
		||||
import java.util.Random;
 | 
			
		||||
 | 
			
		||||
public class Main {
 | 
			
		||||
    public static void main(String[] args) {
 | 
			
		||||
        Random rng = new Random(10);
 | 
			
		||||
        AVLDrevo tree = new AVLDrevo();
 | 
			
		||||
        tree.vstavi(8);
 | 
			
		||||
        tree.vstavi(3);
 | 
			
		||||
        tree.vstavi(10);
 | 
			
		||||
        tree.vstavi(1);
 | 
			
		||||
        tree.vstavi(6);
 | 
			
		||||
        tree.vstavi(14);
 | 
			
		||||
        tree.vstavi(6);
 | 
			
		||||
        tree.vsotaVMejah(3, 6);
 | 
			
		||||
        tree.najdi(9);
 | 
			
		||||
        tree.ktiNajmanjsi(2);
 | 
			
		||||
        tree.najnizjiSkupniPredhodnik(1, 6);
 | 
			
		||||
        int min = -1000;
 | 
			
		||||
        int max = 1000;
 | 
			
		||||
        for (int i = 1; i < 2000; i++) {
 | 
			
		||||
            for (int j = 0; j < 10; j++) {
 | 
			
		||||
                tree.vstavi(rng.nextInt(2000) - 1000);
 | 
			
		||||
            }
 | 
			
		||||
            for (int k = 0; k < 4; k++) {
 | 
			
		||||
                for (int j = 0; j < 30; j++) {
 | 
			
		||||
                    tree.izbrisi(rng.nextInt(2000) - 1000);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        tree.premiPregled();
 | 
			
		||||
        for (int i = 1; i < 200; i++) {
 | 
			
		||||
            tree.ktiNajmanjsi(i);
 | 
			
		||||
        }
 | 
			
		||||
        for (int i = 0; i < 100; i++) {
 | 
			
		||||
            int val = rng.nextInt(max - min) + min;
 | 
			
		||||
            tree.vsotaVMejah(val, val + 200);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user