今天有幸徒手在记事本编程,大概完成了上面的算法逻辑,思路还是很简单,大家看看对不对
void main(String[] args){
List<List<TreeNode>> result=new ArrayList<>();
List<List<TreeNode>> paths=findPath(root);
while(paths.size()>0){
List<TreeNode> path=paths[0];
paths.removeAt(0);
//验证是否满足要求
int sumNum=calSum(path);
if(sumNum==22){
result.add(path);
continue;
}
TreeNode lastNode=path.get(path.size()-1);
List<List<TreeNode>> tempPaths=findPath(lastNode);
//移除最后一个
path.removeAt(path.size()-1);
for(List<TreeNode> tempPath:tempPaths){
List<TreeNode> newPath=new ArrayList();
newPath.addRange(path);
newPath.addRange(tempPath);
paths.add(newPath);
}
}
for(List<TreeNode> nodes:result){
String s=StringUtils.join(",",nodes.stream().map(p->p.value).toArray());
System.out.printIn(s);
}
}
int calSum(List<TreeNode> nodes){
int sum=0;
for(TreeNode node:nodes){
sum+=node.value;
}
return sum;
}
void List<List<TreeNode>> findPath(TreeNode root){
List<List<TreeNode>> nodes=new ArrayList();
if(root.left!=null && root.right!=null){
List<TreeNode> leftNodes=new ArrayList();
List<TreeNode> rightNodes=new ArrayList();
leftNodes.add(root);
leftNodes.add(root.left)
rightNodes.add(root);
rightNodes.add(root.right);
nodes.add(leftNodes);
nodes.add(rightNodes);
}
else if(root.left!=null || root.right!=null){
List<TreeNode> tempNodes=new ArrayList();
tempNodes.add(root);
tempNodes.add(root.left!=null?root.left:root.right);
nodes.add(tempNodes);
}
return nodes;
}