java树的查找_Java实现多叉树查找

1 package tree;

2

3 import java.util.List;

4 import java.util.ArrayList;

5 import java.io.Serializable;

6

7 public class TreeNode implements Serializable {

8     private int parentId;

9     private int selfId;

10     protected String nodeName;

11     protected Object obj;

12     protected TreeNode parentNode;

13     protected List childList;

14

15     public TreeNode() {

16         initChildList();

17     }

18

19     public TreeNode(TreeNode parentNode) {

20         this.getParentNode();

21         initChildList();

22     }

23

24     public boolean isLeaf() {

25         if (childList == null) {

26             return true;

27         } else {

28             if (childList.isEmpty()) {

29                 return true;

30             } else {

31                 return false;

32             }

33         }

34     }

35

36     /* 插入一个child节点到当前节点中 */

37     public void addChildNode(TreeNode treeNode) {

38         initChildList();

39         childList.add(treeNode);

40     }

41

42     public void initChildList() {

43         if (childList == null)

44             childList = new ArrayList();

45     }

46

47     public boolean isValidTree() {

48         return true;

49     }

50

51     /* 返回当前节点的父辈节点集合 */

52     public List getElders() {

53         List elderList = new ArrayList();

54         TreeNode parentNode = this.getParentNode();

55         if (parentNode == null) {

56             return elderList;

57         } else {

58             elderList.add(parentNode);

59             elderList.addAll(parentNode.getElders());

60             return elderList;

61         }

62     }

63

64     /* 返回当前节点的晚辈集合 */

65     public List getJuniors() {

66         List juniorList = new ArrayList();

67         List childList = this.getChildList();

68         if (childList == null) {

69             return juniorList;

70         } else {

71             int childNumber = childList.size();

72             for (int i = 0; i < childNumber; i++) {

73                 TreeNode junior = childList.get(i);

74                 juniorList.add(junior);

75                 juniorList.addAll(junior.getJuniors());

76             }

77             return juniorList;

78         }

79     }

80

81     /* 返回当前节点的孩子集合 */

82     public List getChildList() {

83         return childList;

84     }

85

86     /* 删除节点和它下面的晚辈 */

87     public void deleteNode() {

88         TreeNode parentNode = this.getParentNode();

89         int id = this.getSelfId();

90

91         if (parentNode != null) {

92             parentNode.deleteChildNode(id);

93         }

94     }

95

96     /* 删除当前节点的某个子节点 */

97     public void deleteChildNode(int childId) {

98         List childList = this.getChildList();

99         int childNumber = childList.size();

100         for (int i = 0; i < childNumber; i++) {

101             TreeNode child = childList.get(i);

102             if (child.getSelfId() == childId) {

103                 childList.remove(i);

104                 return;

105             }

106         }

107     }

108

109     /* 动态的插入一个新的节点到当前树中 */

110     public boolean insertJuniorNode(TreeNode treeNode) {

111         int juniorParentId = treeNode.getParentId();

112         if (this.parentId == juniorParentId) {

113             addChildNode(treeNode);

114             return true;

115         } else {

116             List childList = this.getChildList();

117             int childNumber = childList.size();

118             boolean insertFlag;

119

120             for (int i = 0; i < childNumber; i++) {

121                 TreeNode childNode = childList.get(i);

122                 insertFlag = childNode.insertJuniorNode(treeNode);

123                 if (insertFlag == true)

124                     return true;

125             }

126             return false;

127         }

128     }

129

130     /* 找到一颗树中某个节点 */

131     public TreeNode findTreeNodeById(int id) {

132         if (this.selfId == id)

133             return this;

134         if (childList.isEmpty() || childList == null) {

135             return null;

136         } else {

137             int childNumber = childList.size();

138             for (int i = 0; i < childNumber; i++) {

139                 TreeNode child = childList.get(i);

140                 TreeNode resultNode = child.findTreeNodeById(id);

141                 if (resultNode != null) {

142                     return resultNode;

143                 }

144             }

145             return null;

146         }

147     }

148

149     /* 遍历一棵树，层次遍历 */

150     public void traverse() {

151         if (selfId < 0)

152             return;

153         print(this.selfId);

154         if (childList == null || childList.isEmpty())

155             return;

156         int childNumber = childList.size();

157         for (int i = 0; i < childNumber; i++) {

158             TreeNode child = childList.get(i);

159             child.traverse();

160         }

161     }

162

163     public void print(String content) {

164         System.out.println(content);

165     }

166

167     public void print(int content) {

168         System.out.println(String.valueOf(content));

169     }

170

171     public void setChildList(List childList) {

172         this.childList = childList;

173     }

174

175     public int getParentId() {

176         return parentId;

177     }

178

179     public void setParentId(int parentId) {

180         this.parentId = parentId;

181     }

182

183     public int getSelfId() {

184         return selfId;

185     }

186

187     public void setSelfId(int selfId) {

188         this.selfId = selfId;

189     }

190

191     public TreeNode getParentNode() {

192         return parentNode;

193     }

194

195     public void setParentNode(TreeNode parentNode) {

196         this.parentNode = parentNode;

197     }

198

199     public String getNodeName() {

200         return nodeName;

201     }

202

203     public void setNodeName(String nodeName) {

204         this.nodeName = nodeName;

205     }

206

207     public Object getObj() {

208         return obj;

209     }

210

211     public void setObj(Object obj) {

212         this.obj = obj;

213     }

214 }