题意:一棵二叉树每个结点包含一个整数,请设计一个算法输出所有满足条件的路径,此路径上的所有结点之和等于给定值,注意此类路径不要求必须从根结点开始。
解题思路:利用先序遍历的递归算法遍历二叉树,采用数组记录遍历的路径,计算从根结点或者是之后的某一个结点开始至当前结点是否能满足条件,若可以则输出,否则返回上一层改变遍历方向重新计算。代码如下:
//一棵二叉树每个结点包含一个整数,设计一个算法输出所有满足条件的路径,
//此路径上的所有结点之和等于给定值,此类路径不要求必须从根结点开始
#include<iostream>
#include<vector>
using namespace std;
typedef struct BiTreeNode{
int data;
BiTreeNode *lchild;
BiTreeNode *rchild;
}*BiTree;
//从root结点开始向上遍历查找满足条件的路径
//sum结点之和的给定值,buffer记录从当前结点开始向上遍历的路径(此参数为引用),level记录当前结点root所在层次
void findSum(BiTree root, int sum, vector<int> &buffer, int level)
{
if(root==NULL)//递归出口
return;
int tmp=sum;
buffer.push_back(root->data);//记录下当前结点
//从当前结点开始,然后不断遍历与当前结点相连的上一层结点,直到能找到满足条件的路径再输出.
for(int i=level;i>=0;i--)
{
tmp-=buffer[i];
if(tmp==0)//当前路径满足,输出路径
{
for(int j=i;j<=level;j++)
cout<<buffer[j]<<" ";
cout<<endl;
}
}
//从左孩子结点开始向上遍历
findSum(root->lchild,sum,buffer,level+1);
//删除左孩子结点,从右孩子结点开始向上遍历
if(root->lchild)//注意:当左孩子为空时,因为向量没有保存新内容,故不需要删除向量的元素
buffer.pop_back();
findSum(root->rchild,sum,buffer,level+1);
//当右孩子不为空时,返回上一层需要先删除当前元素
if(root->rchild)
buffer.pop_back();
}
void CreateBiTree(BiTree &root)
{
int data;
cin>>data;
if(data==0)//输入为0时表示空结点
root=NULL;
else
{
root=new BiTreeNode();
root->data=data;
CreateBiTree(root->lchild);
CreateBiTree(root->rchild);
}
}
void main()
{
BiTree root=NULL;
CreateBiTree(root);
cout<<"Input end!"<<endl;
int sum=5;//给定值
vector<int> buffer;
findSum(root,sum,buffer,0);//从根结点即第0层开始
}