优化 tree 泛型

This commit is contained in:
如梦技术 2018-12-27 11:29:51 +08:00
parent d40e743be9
commit a9dcbaacec
2 changed files with 52 additions and 47 deletions

View File

@ -23,11 +23,14 @@ import java.util.List;
*
* @author zhuangqian
*/
public class ForestNodeManager {
public class ForestNodeManager<T extends INode> {
private List<INode> list;// 森林的所有节点
/**
* 森林的所有节点
*/
private List<T> list;
public ForestNodeManager(List<INode> items) {
public ForestNodeManager(List<T> items) {
list = items;
}
@ -39,9 +42,10 @@ public class ForestNodeManager {
*/
public INode getTreeNodeAT(int id) {
for (INode forestNode : list) {
if (forestNode.getId() == id)
if (forestNode.getId() == id) {
return forestNode;
}
}
return null;
}
@ -50,12 +54,13 @@ public class ForestNodeManager {
*
* @return 树的根节点集合
*/
public List<INode> getRoot() {
List<INode> roots = new ArrayList<>();
for (INode forestNode : list) {
if (forestNode.getParentId() == 0)
public List<T> getRoot() {
List<T> roots = new ArrayList<>();
for (T forestNode : list) {
if (forestNode.getParentId() == 0) {
roots.add(forestNode);
}
}
return roots;
}

View File

@ -31,9 +31,9 @@ public class ForestNodeMerger {
* @param items 节点域
* @return 多棵树的根节点集合
*/
public static List<INode> merge(List<INode> items) {
ForestNodeManager forestNodeManager = new ForestNodeManager(items);
for (INode forestNode : items) {
public static <T extends INode> List<T> merge(List<T> items) {
ForestNodeManager<T> forestNodeManager = new ForestNodeManager<>(items);
for (T forestNode : items) {
if (forestNode.getParentId() != 0) {
INode node = forestNodeManager.getTreeNodeAT(forestNode.getParentId());
node.getChildren().add(forestNode);