Java JTree_3

/** * */ package com.han; import java.awt.BorderLayout; import java.awt.Container; import java.awt.GridLayout; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.util.Enumeration; import javax.swing.JButton; import javax.swing.JFrame; import javax.swing.JPanel; import javax.swing.JTree; import javax.swing.tree.DefaultMutableTreeNode; import javax.swing.tree.DefaultTreeModel; import javax.swing.tree.TreePath; /** * This class is to enumerate all the nodes of a tree in different ways ... * 遍历树节点 * @author HAN * */ public class JTree_3 extends JFrame { /** * */ private static final long serialVersionUID = -4405016910342910815L; DefaultMutableTreeNode root; /** * */ public JTree_3() { root = new DefaultMutableTreeNode("ROOT"); DefaultMutableTreeNode nodeA = new DefaultMutableTreeNode("nodeA"); DefaultMutableTreeNode nodeB = new DefaultMutableTreeNode("nodeB"); DefaultMutableTreeNode nodeC = new DefaultMutableTreeNode("nodeC"); root.add(nodeA); root.add(nodeB); root.add(nodeC); DefaultMutableTreeNode nodeAA = new DefaultMutableTreeNode("nodeAA"); DefaultMutableTreeNode nodeAB = new DefaultMutableTreeNode("nodeAB"); nodeA.add(nodeAA); nodeA.add(nodeAB); DefaultMutableTreeNode nodeCA = new DefaultMutableTreeNode("nodeCA"); DefaultMutableTreeNode nodeCB = new DefaultMutableTreeNode("nodeCB"); DefaultMutableTreeNode nodeCC = new DefaultMutableTreeNode("nodeCC"); nodeC.add(nodeCA); nodeC.add(nodeCB); nodeC.add(nodeCC); DefaultMutableTreeNode nodeCBA = new DefaultMutableTreeNode("nodeCBA"); DefaultMutableTreeNode nodeCBB = new DefaultMutableTreeNode("nodeCBB"); nodeCB.add(nodeCBA); nodeCB.add(nodeCBB); DefaultTreeModel defaultTreeModel = new DefaultTreeModel(root); JTree tree = new JTree(defaultTreeModel); @SuppressWarnings("rawtypes") Enumeration enumeration = root.preorderEnumeration(); while (enumeration.hasMoreElements()) { DefaultMutableTreeNode node = (DefaultMutableTreeNode) enumeration.nextElement(); TreePath treePath = new TreePath(node.getPath()); tree.expandPath(treePath); } Container container = getContentPane(); container.add(tree, BorderLayout.WEST); // Dimension dimension = container.getPreferredSize(); JPanel panel = new JPanel(new GridLayout(5, 1, 10, 10)); JButton b1 = new JButton("按前序遍历"); JButton b2 = new JButton("按后序遍历"); JButton b3 = new JButton("以广度优先遍历"); JButton b4 = new JButton("以深度优先遍历"); JButton b5 = new JButton("遍历直属子节点"); panel.add(b1); panel.add(b2); panel.add(b3); panel.add(b4); panel.add(b5); container.add(panel, BorderLayout.CENTER); b1.addActionListener(new ActionListener() { @Override public void actionPerformed(ActionEvent e) { // TODO Auto-generated method stub enumerate("按前序遍历"); } }); b2.addActionListener(new ActionListener() { @Override public void actionPerformed(ActionEvent e) { // TODO Auto-generated method stub enumerate("按后序遍历"); } }); b3.addActionListener(new ActionListener() { @Override public void actionPerformed(ActionEvent e) { // TODO Auto-generated method stub enumerate("以广度优先遍历"); } }); b4.addActionListener(new ActionListener() { @Override public void actionPerformed(ActionEvent e) { // TODO Auto-generated method stub enumerate("以深度优先遍历"); } }); b5.addActionListener(new ActionListener() { @Override public void actionPerformed(ActionEvent e) { // TODO Auto-generated method stub enumerate("遍历直属子节点"); } }); } private void enumerate(String mode) { @SuppressWarnings("rawtypes") Enumeration enumeration; if (mode.equals("按前序遍历")) { enumeration = root.preorderEnumeration(); }else if (mode.equals("按后序遍历")) { enumeration = root.postorderEnumeration(); }else if (mode.equals("以广度优先遍历")) { enumeration = root.breadthFirstEnumeration(); }else if (mode.equals("以深度优先遍历")) { enumeration = root.depthFirstEnumeration(); }else { enumeration = root.children(); } System.out.println(mode + " :"); System.out.println("_______________________________"); while (enumeration.hasMoreElements()) { DefaultMutableTreeNode node = (DefaultMutableTreeNode) enumeration.nextElement(); for (int i = 0; i < node.getLevel(); i++) { // 根据节点的级别输出占位符 System.out.print("----"); } System.out.println(node.getUserObject()); // 输出节点标签 } System.out.println(); } /** * @param args */ public static void main(String[] args) { JTree_3 instance = new JTree_3(); instance.setTitle("遍历树节点"); instance.setVisible(true); instance.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); instance.pack(); } }