溫馨提示×

Java怎么生成樹形結(jié)構(gòu)數(shù)據(jù)

小億
98
2023-11-18 16:24:52
欄目: 編程語言

Java中生成樹形結(jié)構(gòu)數(shù)據(jù)可以使用多種方法,下面列舉了兩種常用的方法:

方法一:使用遞歸實現(xiàn)

class TreeNode {
    int val;
    List<TreeNode> children;

    public TreeNode(int val) {
        this.val = val;
        this.children = new ArrayList<>();
    }
}

public class TreeGenerator {
    public static TreeNode generateTree(int[] nums, int rootIndex) {
        if (rootIndex >= nums.length) {
            return null;
        }
        
        TreeNode root = new TreeNode(nums[rootIndex]);
        
        int leftChildIndex = 2 * rootIndex + 1;
        int rightChildIndex = 2 * rootIndex + 2;
        
        root.children.add(generateTree(nums, leftChildIndex));
        root.children.add(generateTree(nums, rightChildIndex));
        
        return root;
    }
    
    public static void main(String[] args) {
        int[] nums = {1, 2, 3, 4, 5, 6, 7};
        
        TreeNode root = generateTree(nums, 0);
        
        // 打印樹的結(jié)構(gòu)
        printTree(root, 0);
    }
    
    public static void printTree(TreeNode root, int level) {
        if (root == null) {
            return;
        }
        
        for (int i = 0; i < level; i++) {
            System.out.print("\t");
        }
        
        System.out.println(root.val);
        
        for (TreeNode child : root.children) {
            printTree(child, level + 1);
        }
    }
}

方法二:使用隊列實現(xiàn)

class TreeNode {
    int val;
    List<TreeNode> children;

    public TreeNode(int val) {
        this.val = val;
        this.children = new ArrayList<>();
    }
}

public class TreeGenerator {
    public static TreeNode generateTree(int[] nums) {
        if (nums.length == 0) {
            return null;
        }
        
        TreeNode root = new TreeNode(nums[0]);
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        
        int index = 1;
        while (!queue.isEmpty() && index < nums.length) {
            TreeNode currNode = queue.poll();
            
            // 左孩子
            if (index < nums.length) {
                TreeNode leftChild = new TreeNode(nums[index++]);
                currNode.children.add(leftChild);
                queue.offer(leftChild);
            }
            
            // 右孩子
            if (index < nums.length) {
                TreeNode rightChild = new TreeNode(nums[index++]);
                currNode.children.add(rightChild);
                queue.offer(rightChild);
            }
        }
        
        return root;
    }
    
    public static void main(String[] args) {
        int[] nums = {1, 2, 3, 4, 5, 6, 7};
        
        TreeNode root = generateTree(nums);
        
        // 打印樹的結(jié)構(gòu)
        printTree(root, 0);
    }
    
    public static void printTree(TreeNode root, int level) {
        if (root == null) {
            return;
        }
        
        for (int i = 0; i < level; i++) {
            System.out.print("\t");
        }
        
        System.out.println(root.val);
        
        for (TreeNode child : root.children) {
            printTree(child, level + 1);
        }
    }
}

以上兩種方法都可以根據(jù)給定的數(shù)組生成樹形結(jié)構(gòu)數(shù)據(jù),并且可以通過遞歸或者隊列的方式進行遍歷和打印。

0