java实现树的节点之间距离;向下递归,注意找到一个满足条件的节点要break

package com.zhangle.arithmetic;


import java.util.ArrayList;
import java.util.List;


public class NodeofTreeDistance {
static class Tree {
private List<Node> list = new ArrayList<Node>();


/**
* 内部类

* @author buaa-pc

*/
class Node {
String data;
String parent;
}


public void add(String parent, String child) {
Node n = new Node();
n.data = child;
n.parent = parent;
list.add(n);

}


/**
* 获取父节点

* @param x
* @return
*/
public String getParent(String x) {
for (int i = 0; i < list.size(); i++) {
if (list.get(i).data.equals(x)) {// 比较内容
return list.get(i).parent;
}
}
return null;
}


public List<String> getChild(String x) {
List<String> newList = new ArrayList<String>();
for (int i = 0; i < list.size(); i++) {
if (list.get(i).parent.equals(x)) {
newList.add(list.get(i).data);
}


}
return newList;
}

public int distance(Tree t,String x,String y) {
int distance=0;
for (int i = 0; i < t.getChild(x).size(); i++) {

if (t.getChild(x).toString().contains(y)) {
distance=1;
break;
}
else {
String city=t.getChild(x).get(i);
int z=distance(t,t.getChild(x).get(i),y);
if (z!=0) {//这里判断下,如果先前有节点已经包含了就跳出来
distance=z+1;
break;
}


}
}

return distance;
}
}


/**
* @param args
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
Tree a = new Tree();
a.add("世界", "欧洲");
a.add("世界", "亚洲");
a.add("世界", "美洲");
a.add("亚洲", "中国");
a.add("亚洲", "日本");
a.add("亚洲", "韩国");
a.add("中国", "北京");
a.add("中国", "江苏");
a.add("中国", "上海");
a.add("中国", "安徽");
//求世界到安徽的节点距离
System.out.println("世界与安徽的距离为"+a.distance(a, "世界","安徽"));
//求亚洲到安徽的节点距离
System.out.println("世界与安徽的距离为"+a.distance(a, "亚洲","安徽"));
}


}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
/** * 根据等级查询类目 * * @param level * @return */ @Override public List queryCategoryTree(Integer level) { //查询当前级别下类目 List list = categoryDAO.list(level); //组装好的类目,返回前端 List categoryTree = new ArrayList(); //所有类目 List allDTOList = new ArrayList(); if (CollectionUtils.isEmpty(list)) { return categoryTree; } for (CategoryDO categoryDO : list) { allDTOList.add(new CategoryTreeDTO().convertDOToDTO(categoryDO)); } //当前等级类目 categoryTree = allDTOList.stream().filter(dto -> level.equals(dto.getLevel())).collect(Collectors.toList()); for (CategoryTreeDTO categoryTreeDTO : categoryTree) { //组装类目为结构 assembleTree(categoryTreeDTO, allDTOList,Constants.CATEGORY_MAX_LEVEL - level); } return categoryTree; } /** * 组装 * * @param categoryTreeDTO * @param allList * @param remainRecursionCount 剩余递归次数 * @return */ public CategoryTreeDTO assembleTree(CategoryTreeDTO categoryTreeDTO, List allList, int remainRecursionCount) { remainRecursionCount--; //最大递归次数不超过Constants.CATEGORY_MAX_LEVEL-level次,防止坏数据死循环 if(remainRecursionCount < 0){ return categoryTreeDTO; } String categoryCode = categoryTreeDTO.getCategoryCode(); Integer level = categoryTreeDTO.getLevel(); //到达最后等级返回 if (Constants.CATEGORY_MAX_LEVEL == level) { return categoryTreeDTO; } //子类目 List child = allList.stream().filter(a -> categoryCode.equals(a.getParentCode())).collect(Collectors.toList()); if (null == child) { return categoryTreeDTO; } categoryTreeDTO.setChildren(child); //组装子类目 for (CategoryTreeDTO dto : child) { assembleTree(dto, allList,remainRecursionCount); } return categoryTreeDTO; }
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值