给定一个二叉树和一个目标和,求节点路径

image.png

今天有幸徒手在记事本编程,大概完成了上面的算法逻辑,思路还是很简单,大家看看对不对

void main(String[] args){
    List> result=new ArrayList<>();

    List> paths=findPath(root);
    while(paths.size()>0){
        List 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> tempPaths=findPath(lastNode);
        //移除最后一个
        path.removeAt(path.size()-1);
        for(List tempPath:tempPaths){
            List newPath=new ArrayList();
            newPath.addRange(path);
            newPath.addRange(tempPath);
            paths.add(newPath);
        }
    }
    
    for(List nodes:result){
        String s=StringUtils.join(",",nodes.stream().map(p->p.value).toArray());
        System.out.printIn(s);
    }
}
int calSum(List nodes){
    int sum=0;
    for(TreeNode node:nodes){
        sum+=node.value;
    }
    return sum;
}
void List> findPath(TreeNode root){
    List> nodes=new ArrayList();
    if(root.left!=null && root.right!=null){
        List leftNodes=new ArrayList(); 
        List 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 tempNodes=new ArrayList(); 
        tempNodes.add(root);
        tempNodes.add(root.left!=null?root.left:root.right);
        
        nodes.add(tempNodes);
    }
    return nodes;
}

你可能感兴趣的:(给定一个二叉树和一个目标和,求节点路径)